/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 vote158 views1 answer
Default ordering django
Python
0 vote194 views1 answer
Tabpy Configuration file with custom settings
Python
0 vote152 views1 answer
How to take input in python as string and convert into integer list
Python
57 votes171 views6 answers
Print current date and time in python
Python
2 votes160 views1 answer
Fyit download
Python
0 vote172 views1 answer
Python get user of pid
Python
1 vote170 views1 answer
Python convert querydict to dict
Python
0 vote165 views1 answer
Python turn seconds into zulu time
Python
0 vote194 views1 answer
Activate inherit function django
Python Django
12 votes170 views2 answers
Isinstance float or int
Python