/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 vote178 views1 answer
Make value of two tables unique django
Python
2 votes170 views2 answers
What is role of ALLOWED_HOSTs in Django
Python
0 vote165 views1 answer
Python selenium firefox handle ssl bypass
Python
2 votes161 views1 answer
How to launch jupyter notebook from cmd
Python
0 vote155 views1 answer
Networkx node attribute from a dataframe
Python
0 vote210 views1 answer
Styling filter form django
Python Django
0 vote174 views1 answer
How to get pytroch model layer name
Python
7 votes169 views4 answers
XLRDError: Excel xlsx file; not supported
Python
0 vote149 views2 answers
If something in something python example
Python
0 vote159 views1 answer
How to use wbtools in python
Python