/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 vote207 views1 answer
"api_view" is not defined django
Python
0 vote251 views1 answer
How to close a webpage using selenium driver python
Python
0 vote301 views1 answer
How to numbered jupyter notebook
Python
0 vote248 views2 answers
Python manual elif
Python
3 votes255 views2 answers
Postgres Database Django
Python Django
11 votes269 views6 answers
BeautifulSoup
Python
0 vote299 views1 answer
Django on delete set default
Python Django
186 votes338 views6 answers
How to read file in python
Python
2 votes241 views2 answers
Python printing date
Python
2 votes256 views1 answer
Python check if character is letter
Python