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 vote86 views1 answer
Remove nas from dataframe in python
Python
0 vote142 views1 answer
Resize cmd using python
Python
1 vote120 views2 answers
Rank function in pandas
Python
0 vote113 views1 answer
Axes increase fonsize of values python
Python
5 votes152 views3 answers
How to add images in hml while using flask
Python Flask
0 vote116 views1 answer
Numpy generlized ufunc
Python
0 vote120 views1 answer
Python render_template
Python
1 vote126 views1 answer
Change color of butto in thkinter
Python
4 votes111 views1 answer
F-string ponto decimal python
Python
6 votes126 views2 answers
Python check if folder is empty
Python