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

12 votes268 views5 answers
Append python
Python
5 votes246 views3 answers
Render to response django
Python Django
0 vote251 views2 answers
Get last file in directory python
Python
1 vote237 views1 answer
Python print percent sign
Python
15 votes233 views3 answers
Python - gropuby based on 2 variabels
Python
1 vote245 views1 answer
While not equal python
Python
52 votes284 views6 answers
Sorting a dictionary in python
Python
125 votes253 views6 answers
How to take input from user in python
Python
17 votes228 views4 answers
How to get int input in python
Python
28 votes223 views2 answers
Python print without new lines
Python