/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 vote95 views1 answer
Python turn list of strings into list of doubles
Python
0 vote91 views1 answer
Truc python
Python
1 vote92 views1 answer
Hoe maak je machten in python
Python
85 votes107 views6 answers
For python
Python
2 votes57 views1 answer
Check case string python
Python
0 vote98 views1 answer
Check if digit or alphabet
Python
0 vote97 views1 answer
How to see if something is in a class in python
Python
5 votes95 views4 answers
C code to python code converter online
Python
0 vote98 views1 answer
Tkinter canas can you use other fonts
Python
0 vote106 views1 answer
Tkinter get child in frame
Python