/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

2 votes172 views2 answers
Python program to find first n prime numbers
Python
0 vote136 views1 answer
Install python setuptools ubuntu
Python
0 vote138 views1 answer
Addind scheduling of tasks to pyramid python app
Python
2 votes115 views1 answer
Pandas pass two columns to function
Python
3 votes108 views2 answers
How many columns have null values present in them? in pandas
Python
35 votes113 views2 answers
Pwd python
Python
0 vote112 views1 answer
Register template tag django
Python
0 vote115 views1 answer
How to count the appearance of number or string in a list python
Python
4 votes102 views1 answer
Python data structures
Python
1 vote114 views1 answer
How to locate image using pyautogui
Python