/Python/Pytorch tensor add one dimension

Pytorch tensor add one dimension

# ADD ONE DIMENSION: .unsqueeze(dim)

my_tensor = torch.tensor([1,3,4])
# tensor([1,3,4])

my_tensor.unsqueeze(0)
# tensor([[1,3,4]])

my_tensor.unsqueeze(1)
# tensor([[1],
#         [3],
#         [4]])

Similar codes

0 vote178 views1 answer
How to preserve white space when joining an array python
Python
107 votes146 views6 answers
Python do while loop
Python
0 vote144 views1 answer
Add prefix to names in directory python
Python
0 vote140 views1 answer
Solidity compiler for python
Python
5 votes144 views5 answers
Nmap
Python
8 votes150 views4 answers
Python list prime numbers
Python
4 votes164 views3 answers
How to duplicate a list in python
Python
0 vote181 views1 answer
String to date in BQ
Python
3 votes150 views1 answer
Open cv read webcam
Python
2 votes298 views4 answers
Streamlit python install
Python