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 votes146 views1 answer
Jupyter notebook pass python variable to shell
Python
3 votes147 views1 answer
Discord py on ready
Python
9 votes149 views2 answers
Get a list of all files python
Python
0 vote291 views1 answer
Python args description multiple lines
Python
8 votes192 views2 answers
What does class meta do in django
Python Django
0 vote155 views1 answer
1024x768
Python
0 vote140 views1 answer
Using pickle to create binary files
Python
0 vote144 views1 answer
Get one from dataloader
Python
-1 vote151 views1 answer
Extract specific key values from python dictionary
Python
1 vote154 views1 answer
Incognito mode in selenium
Python