/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

268 votes503 views0 answer
Last element of list python
Python
-1 vote219 views1 answer
Filter groupby pandas
Python
0 vote194 views1 answer
Long press selenium python
Python
1 vote242 views2 answers
Filter titlecase django
Python Django
-2 votes222 views1 answer
Random name generator in python
Python
0 vote182 views1 answer
Torch.utils.data.random_split(dataset, lengths)
Python
0 vote187 views1 answer
Pyspark rdd sort by value descending
Python
8 votes169 views1 answer
String startswith python
Python
0 vote184 views1 answer
Firebase python flask
Python
1 vote190 views2 answers
Install fasttext python
Python