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

2 votes134 views1 answer
Python pandas apply to one column
Python
0 vote144 views1 answer
Dependency inversion
Python
2 votes185 views1 answer
How to set google chrome as default browser when coding with python using webbroiwser module
Python
5 votes159 views4 answers
Ipython play sound
Python
0 vote164 views1 answer
Re mobile no validate python
Python
2 votes140 views1 answer
Pandas reset row indices
Python
0 vote141 views1 answer
Tweepy aut code
Python
1 vote135 views1 answer
No such table: django_session admin
Python
70 votes129 views6 answers
What is python virtual environment
Python
0 vote143 views1 answer
Coercion python
Python