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

89 votes285 views6 answers
Import csv file in python
Python
0 vote249 views1 answer
Admin site
Python
8 votes289 views6 answers
Combine to lists python
Python
46 votes257 views6 answers
Round to decimal places python
Python
1 vote251 views1 answer
How to split a text column into two separate columns?
Python
21 votes278 views5 answers
Format string to 2 decimal places python
Python
5 votes258 views2 answers
Write string to file python
Python
0 vote274 views1 answer
Union dataframe pyspark
Python
23 votes261 views4 answers
Tkinter background color
Python
2 votes262 views1 answer
Hyperlinks in jupyter notebook
Python