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

4 votes163 views1 answer
Dropdown in tkinter
Python
3 votes158 views3 answers
Django.db.utils.OperationalError: no such table:
Python
0 vote156 views1 answer
Data dictionary python into numpy
Python
0 vote157 views1 answer
Word embedding python
Python
0 vote162 views2 answers
Pandas agg rename columns
Python
0 vote183 views1 answer
How to add multiple items in a list in python
Python
31 votes185 views4 answers
Python random generator from list
Python
2 votes162 views1 answer
Basic String Functions in python
Python
0 vote163 views1 answer
Pandas find list based on condition
Python
1 vote180 views1 answer
'Polygon' object has no property 'normed'
Python