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 vote133 views1 answer
Python hash md5 unicode
Python
0 vote160 views1 answer
How to form .cleaned data in class based views in django
Python Django
0 vote154 views2 answers
How to find unique sublist in list in python
Python
0 vote131 views1 answer
Python default function value
Python
6 votes133 views1 answer
Change dataframe value by index
Python
0 vote142 views1 answer
Raise keyerror(key) from none os.environ
Python
0 vote166 views1 answer
How to run mac terminal from python script
Python
1 vote150 views1 answer
Python delete elements from list / range
Python
0 vote133 views1 answer
Monkey patching in python
Python
0 vote130 views1 answer
Label with list comprehension python
Python