Pytorch squeeze

x = torch.zeros(2, 1, 2, 1, 2)
x.size()
>>> torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x) # remove 1
y.size()
>>> torch.Size([2, 2, 2])

y = torch.squeeze(x, 0)
y.size()
>>> torch.Size([2, 1, 2, 1, 2])

y = torch.squeeze(x, 1)
y.size()
>>> torch.Size([2, 2, 1, 2])

Source: pytorch.org

Similar codes

3 votes138 views3 answers
Kadane's algorithm
Python
1 vote333 views1 answer
Replace space with . pyhton
Python
0 vote131 views1 answer
Matplotlib create histogram edge color
Python
0 vote137 views1 answer
Procfile heroku python example
Python
8 votes158 views3 answers
Which python mac
Python
0 vote132 views1 answer
Python edit global variable in function
Python
0 vote146 views1 answer
Vim run python current file
Python
2 votes138 views3 answers
Get queryset
Python
13 votes144 views4 answers
If list item in string python
Python
0 vote143 views2 answers
Cronometer python
Python