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

0 vote194 views1 answer
From Player import Player
Python
0 vote199 views1 answer
Pandas set column to value using mask
Python
0 vote195 views1 answer
Radio button selected event
Python
0 vote209 views1 answer
How do i select a range of columns by index
Python
2 votes205 views2 answers
Forrest Gump
Python
10 votes208 views5 answers
List of dict to dataframe
Python
9 votes240 views2 answers
How to fill nan values with mean in pandas
Python
9 votes195 views3 answers
How to convert lower case to upper case in python
Python
45 votes226 views3 answers
Python get day month year
Python
1 vote218 views1 answer
Google colab save faild
Python