/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 vote246 views1 answer
Sympy.diff
Python
0 vote224 views1 answer
Quoto x discord selfbot
Python
-3 votes252 views6 answers
Python 3.3 release date
Python
4 votes264 views4 answers
Python pandas change column values to all caps
Python
0 vote252 views1 answer
Python OrderedDict - LRU
Python
3 votes287 views1 answer
Code snippet
Python
0 vote288 views1 answer
Python Entry default text
Python
0 vote256 views1 answer
Python geodata visualize
Python
13 votes235 views3 answers
How to change character in string python
Python
1 vote216 views1 answer
Python check if input() gives error
Python