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

2 votes202 views3 answers
Select rows in python
Python
0 vote158 views1 answer
Create_polygon tkinter
Python
0 vote184 views1 answer
Logout from linux using python
Python
0 vote194 views1 answer
How to sum a column in csv python using list in python
Python
0 vote230 views1 answer
Insert-cells-in-empty-pandas-dataframe
Python
1 vote180 views1 answer
Add year to id django
Python
0 vote198 views1 answer
Import numpy import cv2 cv2.imshow('image',img) cv2.waitKey(0)
Python
0 vote221 views1 answer
Free the Bunny Prisoners
Python
0 vote198 views1 answer
Add header info in django response
Python Django
21 votes190 views5 answers
Base64 decode python
Python