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

0 vote122 views1 answer
How to find the summation of all the values in a tuple python
Python
0 vote116 views1 answer
Matplotlib add text to axes
Python
0 vote120 views1 answer
Count upercase
Python
0 vote116 views4 answers
Python call function in class
Python
0 vote113 views1 answer
Binarize array python
Python
1 vote112 views1 answer
Flatten a 2d array python
Python
1 vote114 views1 answer
Pip install qrcode python
Python
0 vote118 views1 answer
Codeforces 677a python solution
Python
0 vote123 views1 answer
Numpy mask without losing shape
Python
0 vote115 views1 answer
Python detect color on screen
Python