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

80 votes270 views6 answers
Python code to exe file
Python
0 vote248 views1 answer
Python loop array 0,101/100
Python
0 vote296 views1 answer
Calculate Euclidean Distance in Python using norm()
Python
2 votes267 views3 answers
Condition python
Python
31 votes298 views4 answers
Python for loop counter
Python
44 votes263 views6 answers
Sklearn python install
Python
0 vote299 views1 answer
Qt line edit set text python
Python
4 votes250 views3 answers
Print variable string python
Python
0 vote262 views1 answer
How to create list of objects in python
Python
1 vote261 views1 answer
Tkinter app icon
Python