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 vote264 views1 answer
Python assert is datetime
Python
0 vote280 views1 answer
How remove name of index pandas
Python
2 votes283 views2 answers
Python isdigit negative
Python
2 votes221 views1 answer
Python text to ascii art
Python
13 votes275 views1 answer
Change background color of tkinter
Python
0 vote287 views1 answer
Python Sorted Word frequency count
Python
0 vote261 views1 answer
Webhook logger python
Python
0 vote314 views1 answer
Python simple number formatting samples
Python
1 vote294 views1 answer
Add background image in django uploaded file
Python
1 vote265 views1 answer
R vs python
Python