/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

1 vote176 views1 answer
Formula for compounding interest in python
Python
0 vote177 views1 answer
Seaborn dot plot hue
Python
0 vote172 views1 answer
Tkinter auto resize height
Python
0 vote221 views1 answer
I have two versions of python installed mac
Python
3 votes191 views2 answers
Python execute bat file
Python
145 votes203 views5 answers
How to generate a random number python
Python
0 vote171 views1 answer
Numpy collapse last dimension
Python
1 vote207 views1 answer
Python sum attribute in list
Python
0 vote174 views1 answer
A = [ int(i) for i in range(100, 104)] list python
Python
1 vote214 views1 answer
How to add a cooment in python
Python