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 vote141 views1 answer
400/15
Python
0 vote165 views1 answer
How to get the highest number of an input in python
Python
0 vote152 views1 answer
Numpy get index of nan
Python
4 votes140 views1 answer
Python return -1
Python
0 vote216 views1 answer
Python - How To Convert Bytearray to String
Python
12 votes158 views6 answers
How to make a list python
Python
0 vote146 views1 answer
Numpy.where() for substring
Python
1 vote161 views1 answer
Postgres django settings
Python
0 vote153 views1 answer
Run python with options
Python
2 votes142 views2 answers
How to find the longest string in a list in python
Python