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 vote109 views1 answer
Python sort with comparator
Python
4 votes101 views3 answers
Row wise mean pandas
Python
0 vote94 views2 answers
Adding labels to histogram bars in matplotlib
Python
0 vote94 views1 answer
Compter des valeur consecutives en python
Python
0 vote90 views1 answer
Win64pyinstaller
Python
0 vote90 views1 answer
Python sum 1-50
Python
0 vote102 views1 answer
From random input python
Python
1 vote131 views1 answer
How to view the whole dataset in jupyternotebook
Python
9 votes93 views6 answers
Module in python
Python
19 votes125 views6 answers
How to import turtle in python
Python