/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 vote172 views1 answer
Django email verification
Python Django
72 votes126 views4 answers
Check python version mac
Python
1 vote134 views1 answer
Acess nvidia from docker compose
Python
-1 vote145 views1 answer
# get the largest number in a list and print its indexes
Python
0 vote179 views1 answer
Django models for existing database
Python Django
0 vote181 views1 answer
Django update replace regex
Python Django
18 votes128 views4 answers
Edit path variable using python
Python
0 vote144 views1 answer
%d%m%y python
Python
2 votes139 views1 answer
How do I mock an uploaded file in django?
Python
3 votes144 views2 answers
Python use variable in another file
Python