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 votes245 views2 answers
Pandas - from umeric to string
Python
0 vote214 views1 answer
Run file as administrator python
Python
6 votes191 views2 answers
How to install python2 in ubuntu
Python
0 vote224 views1 answer
Django composer
Python Django
79 votes231 views6 answers
How to create a virtual environment in python 3
Python
18 votes214 views6 answers
Gui in python
Python
1 vote459 views1 answer
TypeError: unhashable type: 'list' python dictionary
Python
2 votes218 views1 answer
Python system of nonlinear equations
Python
1 vote201 views1 answer
Python close all plot figures
Python
7 votes245 views2 answers
Python cd to script directory
Python