/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

2 votes277 views1 answer
Pandas change last row
Python
0 vote228 views1 answer
All python versions
Python
0 vote250 views1 answer
Python list of possible paths
Python
3 votes324 views1 answer
Access the value in settings django
Python Django
4 votes248 views1 answer
Data model
Python
0 vote250 views1 answer
Data where values in column starts with particular value
Python
3 votes258 views2 answers
What day i s it
Python
2 votes289 views1 answer
Version of scikit learn
Python
1 vote290 views1 answer
Create empty numpy array without shape
Python
4 votes249 views6 answers
If statement python
Python