/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

4 votes389 views1 answer
Dataframe to csv without ids
Python
6 votes345 views2 answers
Python datetime get all days between two dates
Python
1 vote332 views1 answer
Selenium select element by id
Python
0 vote306 views1 answer
Locate certificate path python
Python
11 votes348 views6 answers
Pandas join two columns
Python
0 vote366 views1 answer
Customise django admin edit model button in field
Python Django
0 vote342 views1 answer
Remove stopwords from list of strings python
Python
1 vote350 views2 answers
Change base python
Python
1 vote365 views2 answers
How to swap two variables without using third variable python
Python
9 votes352 views6 answers
Activate virtual environment python windows 10
Python