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

0 vote234 views1 answer
Dev.to flask heroku
Python
0 vote162 views1 answer
True Positive, True Negative, False Positive, False Negative in scikit learn
Python
5 votes182 views1 answer
Split and join in python
Python
0 vote174 views1 answer
Pyqt5 image change size
Python
0 vote172 views1 answer
Python apply function to list
Python
1 vote261 views1 answer
Python split pdf pages
Python
0 vote168 views1 answer
Odd or even checker
Python
4 votes251 views1 answer
Python django shell command
Python Django
-2 votes167 views2 answers
String index
Python
1 vote186 views2 answers
Valid parentheses
Python