/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

4 votes338 views2 answers
Python check if exe is running
Python
4 votes314 views1 answer
Np.random.float
Python
0 vote350 views1 answer
Fastapi upload file save
Python
8 votes349 views2 answers
How to earse special chrat¥cter from string in python
Python
1 vote317 views1 answer
Intersection of dataframes based on column
Python
0 vote373 views1 answer
Tuple: Create tuple
Python
11 votes325 views1 answer
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xe2 in position 10: invalid continuation byte
Python
7 votes361 views2 answers
How to generate requirements.txt django
Python Django
28 votes320 views2 answers
Python print without new lines
Python
0 vote338 views1 answer
Minimumm sort tryed python
Python