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 vote320 views1 answer
Python mean ndarray
Python
0 vote316 views1 answer
Python web server oneliner
Python
1 vote311 views1 answer
Random pick any file from directory python
Python
5 votes307 views1 answer
How to print for loop in same line in python
Python
0 vote319 views1 answer
Pip ne marche pas
Python
1 vote384 views1 answer
Tkinter hover button
Python
0 vote337 views1 answer
Best way to calculate tax python
Python
0 vote331 views1 answer
Genrate random code in django
Python
0 vote321 views1 answer
Python get volume free space
Python
3 votes320 views2 answers
Remove duplicates from sorted array
Python