/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

84 votes362 views2 answers
Python - save file
Python
1 vote359 views1 answer
Test cuda pytorch
Python
0 vote343 views1 answer
Same quotes in a quotes
Python
0 vote352 views1 answer
Django check if get parameter exists
Python Django
3 votes364 views5 answers
Pandas split column into multiple columns
Python
10 votes327 views3 answers
Gson
Python
0 vote391 views1 answer
Django auto complete light styling
Python Django
0 vote290 views1 answer
Comment enleve les chiffre duplice d une liste python
Python
0 vote324 views1 answer
Series distict
Python
7 votes370 views5 answers
.format python 3
Python