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 vote142 views1 answer
Sum with conditional python
Python
0 vote133 views1 answer
How do i select a range of columns by index
Python
4 votes145 views2 answers
Count characters in string python
Python
0 vote149 views1 answer
Deep clone 2d list python
Python
0 vote152 views3 answers
How to use css in php example
Python
0 vote143 views1 answer
Get resource path python
Python
0 vote141 views1 answer
Ciclo while python
Python
6 votes146 views1 answer
Anaconda python update packages
Python
0 vote153 views1 answer
Create array of specific size python
Python
-1 vote140 views1 answer
How to add one 2d list to another 2d list python
Python