Torch.unsqueze

>>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1,  2,  3,  4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
        [ 2],
        [ 3],
        [ 4]])

Source: pytorch.org

Similar codes

1 vote201 views2 answers
Num1 = int(input('Enter The First No: ')) num2 = int(input('Enter The Second No: ')) print("Answer: "num1+num2)
Python
0 vote195 views1 answer
Aws lambda logging with python logging library
Python
8 votes192 views2 answers
Python iterate list reverse
Python
1 vote196 views1 answer
Stop procedure python
Python
0 vote217 views1 answer
Exponential regression python
Python
1 vote196 views1 answer
Read shp in python
Python
0 vote192 views1 answer
Python getpass save file
Python
1 vote260 views1 answer
Django return only part of string
Python
7 votes257 views3 answers
Python change data type to integer
Python
5 votes198 views1 answer
Python pip yaml
Python