/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 vote114 views1 answer
Printing a range of no one line in python
Python
5 votes124 views3 answers
How to extract zip file using python
Python
4 votes102 views4 answers
Flash messages django
Python
0 vote100 views1 answer
Inverse box-cox transformation python
Python
0 vote100 views10 answers
PEP 428: The pathlib module – object-oriented filesystem paths.
Python
1 vote61 views1 answer
How to create a list of tuples in python
Python
0 vote105 views1 answer
Arrange array in ascending order python
Python
4 votes101 views1 answer
Python extract all numbers from string re
Python
-2 votes124 views4 answers
Django form datepicker
Python
0 vote107 views1 answer
Sqlite python select with parameters
Python