/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

15 votes162 views3 answers
How to get the current year in python
Python
3 votes149 views2 answers
Mac why is python installed in usr and application
Python
5 votes193 views4 answers
Django get or 404
Python Django
14 votes149 views6 answers
How to print hello world in python
Python
2 votes151 views1 answer
Python how to replace a certain string in text
Python
3 votes119 views2 answers
Plot bounds python
Python
0 vote147 views1 answer
Rounding values in pandas dataframe
Python
3 votes210 views1 answer
From _curses import * ModuleNotFoundError: No module named '_curses'
Python
0 vote151 views1 answer
Find where df series is null and print
Python
0 vote170 views1 answer
Python code to convert all keys of dict into lowercase
Python