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

0 vote189 views1 answer
How to get a int from string python
Python
0 vote189 views1 answer
Filter dict by list of keys python
Python
0 vote240 views1 answer
Appending objects to a list contained in a dictionary python
Python
0 vote282 views1 answer
List generation for python
Python
13 votes182 views2 answers
Only keep rows of a dataframe based on a column value
Python
0 vote360 views1 answer
Import flask session
Python Flask
60 votes195 views4 answers
Datetime year python
Python
0 vote204 views1 answer
Destructuring in for loops python
Python
1 vote166 views2 answers
Python3 shebang
Python
1 vote185 views1 answer
Shutil move overwrite
Python