/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 vote258 views1 answer
Networkx - calculate degree per each node
Python
7 votes236 views4 answers
'numpy.ndarray' object is not callable
Python
0 vote283 views1 answer
Reverse relationship in django for one to one field for usage in Django rest serializer
Python
0 vote255 views1 answer
Violajones python opencv
Python
33 votes257 views4 answers
Convert list into integer in python
Python
0 vote283 views1 answer
Python subprocess print stdout while process running
Python
31 votes276 views6 answers
Count list python
Python
0 vote281 views1 answer
How to split a string with newline in python
Python
0 vote275 views1 answer
Sqlite query using string as parameter in python
Python
1 vote276 views1 answer
Python nCr n choose r function
Python