/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 vote257 views2 answers
Fibonacci sequence script python
Python
2 votes277 views1 answer
Django models distinct
Python Django
0 vote258 views1 answer
Herons rule python
Python
0 vote260 views1 answer
Python how to code discord bot kick members
Python
1 vote287 views1 answer
How to get ipconfig from python
Python
0 vote254 views1 answer
Flask return 404
Python
0 vote235 views1 answer
Different types f python loops
Python
0 vote293 views1 answer
Include app in django project
Python Django
2 votes243 views1 answer
Zermelo api
Python
1 vote237 views1 answer
Tkinter new line in text
Python