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 votes307 views2 answers
Python tkinter programming project ideas
Python
4 votes309 views2 answers
Python map input
Python
-1 vote314 views1 answer
Como agregar elementos a un array en python
Python
0 vote333 views2 answers
Where should favicon.ico be placed in flask app
Python
0 vote355 views1 answer
Python request post with json with headers
Python
0 vote370 views1 answer
Screen.onkey python
Python
4 votes309 views1 answer
Server on python with address and port
Python
1 vote364 views1 answer
Change the user agent selenium
Python
3 votes311 views1 answer
Mouse bottom in pygame
Python
1 vote344 views1 answer
No module named 'arabic_reshaper'
Python