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 votes103 views6 answers
K
Python
0 vote73 views1 answer
Unique id on duplicate ids values pandas
Python
0 vote111 views1 answer
Write a python program which accepts the user
Python
-1 vote106 views1 answer
Can only use .dt accessor with datetimelike values
Python
5 votes142 views2 answers
How to reset username and password in django admin
Python Django
19 votes121 views3 answers
How to delete a column from a dataframe in python
Python
3 votes112 views1 answer
Streamlit ssl error
Python
71 votes60 views1 answer
Small python projects for beginners
Python
74 votes113 views9 answers
Online python compiler
Python
1 vote94 views1 answer
_set in django
Python