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

19 votes177 views5 answers
Pandas where retuning NaN
Python
0 vote162 views1 answer
Imblearn RandomOverSampler
Python
1 vote186 views1 answer
Run code with different verions of python
Python
0 vote173 views1 answer
Extract label from tf data
Python
10 votes158 views4 answers
Calculate age python
Python
66 votes166 views6 answers
How to open csv file in python
Python
0 vote173 views2 answers
Python override string class
Python
-2 votes150 views1 answer
Discord python send private message
Python
1 vote163 views1 answer
Python 3 tkinter treeview example
Python
0 vote161 views1 answer
Scrapy create project
Python