/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

1 vote330 views1 answer
Use miraculous with token
Python
0 vote287 views1 answer
Unhapppy man with monwy
Python
0 vote317 views4 answers
Python call function from string
Python
0 vote412 views1 answer
Field 'id' expected a number but got <django.db.models.fields.related_descriptors.create_forward_many_to_many_manager.<locals>.ManyRelatedManager object at 0x7fb02044de80>
Python Django
1 vote328 views1 answer
How to get sum specific columns value in machine learning
Python
29 votes310 views5 answers
Python ls directory
Python
0 vote323 views1 answer
Install python 3.4 mac terminal
Python
10 votes335 views3 answers
Django make migrations
Python
4 votes321 views3 answers
How to convert list into csv in python
Python
0 vote363 views1 answer
Qt line edit set text python
Python