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

1 vote433 views1 answer
Calculate integral python
Python
5 votes376 views6 answers
Valid Palindrome
Python
0 vote335 views1 answer
Time addition in python
Python
0 vote334 views1 answer
Spacy load en
Python
-1 vote379 views1 answer
How to find root value python
Python
0 vote329 views1 answer
How to deploy a file size greater than 100mb on pythonanywhere
Python
0 vote345 views1 answer
Get current module name python
Python
0 vote346 views2 answers
Sorted set in python
Python
0 vote372 views1 answer
Sum of the number in a list in python
Python
0 vote344 views1 answer
Example of Python Inline comments
Python