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

1 vote202 views2 answers
Python file basename
Python
0 vote202 views1 answer
Nested For loop in Python
Python
1 vote213 views1 answer
How to convert an image to ascii art using python
Python
3 votes220 views1 answer
Plotly not showing in jupyter
Python
22 votes216 views4 answers
ModuleNotFoundError: No module named 'sklearn.cross_validation'
Python
0 vote214 views1 answer
Ptyhton json respones
Python
4 votes199 views3 answers
Unable to locate package python-pip
Python
3 votes208 views2 answers
Thread with args python
Python
1 vote207 views1 answer
Python regex get string before character
Python
2 votes213 views1 answer
How to reomve certain row from dataframe pandas
Python