/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

2 votes189 views1 answer
Python default input
Python
8 votes179 views2 answers
Selenium upload file python
Python
-1 vote188 views1 answer
Pandas numpy multiplicar dos columnas segun una condicion
Python
4 votes194 views1 answer
Delete element of a list from another list python
Python
0 vote193 views1 answer
How to run fastapi with code python
Python
0 vote193 views1 answer
Sample one point from distribution python
Python
4 votes185 views2 answers
Python run command and read output
Python
6 votes185 views3 answers
Django filter install
Python
0 vote184 views1 answer
Modules django
Python
0 vote253 views1 answer
Circular import python
Python