/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

10 votes157 views4 answers
Multiply each element in list python
Python
0 vote179 views1 answer
Import models
Python
4 votes167 views2 answers
Python mkdir
Python
0 vote185 views1 answer
Python get number of arguments of a function
Python
3 votes149 views2 answers
Upgrade python to 3.8
Python
0 vote190 views1 answer
Django check user admin
Python Django
5 votes143 views1 answer
Tkiner border
Python
4 votes151 views1 answer
What's the equivalent to System.nanotime in python
Python
0 vote154 views3 answers
Remove string from list in python
Python
2 votes160 views1 answer
Difference between sort and sorted
Python