/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 votes116 views3 answers
Python pandas difference between two data frames
Python
5 votes80 views1 answer
Series vs dataframe
Python
0 vote141 views1 answer
How to remove repeated files from google drive in python
Python
0 vote148 views1 answer
Django 2.2 disable cache settings.STATIC_URL
Python Django
0 vote119 views1 answer
Pandas resample groupby
Python
0 vote142 views1 answer
Python get operating system name
Python
0 vote117 views1 answer
Logging store info to different files
Python
0 vote155 views1 answer
Django rest framework default_authentication_classes
Python Django
0 vote104 views1 answer
OneHotEncoder pyspark
Python
0 vote168 views1 answer
No module named 'bootstrap4' django
Python Django