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

49 votes326 views10 answers
Reddit
Python
25 votes299 views6 answers
Python datetime to timestamp
Python
0 vote303 views1 answer
Add values to pandas plot
Python
0 vote348 views1 answer
WARNING: Ignoring invalid distribution -ip
Python
8 votes286 views1 answer
Python find runtime
Python
0 vote311 views1 answer
Django redirect to external url
Python
3 votes291 views1 answer
How to delete a turtle in python
Python
0 vote328 views1 answer
"json" is not defined
Python
1 vote290 views2 answers
Django timezone settings
Python
12 votes306 views3 answers
Setwd python
Python