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

4 votes283 views2 answers
Python number to array of digits
Python
12 votes276 views3 answers
How to open any application using python
Python
34 votes288 views6 answers
Pandas drop row with nan
Python
-86 votes281 views10 answers
Code
Python
0 vote283 views1 answer
Gitpod how to execute python file
Python
0 vote274 views1 answer
Python how to locate and fill a specific column null values
Python
0 vote337 views1 answer
Flask mongoengine getting started
Python Flask
1 vote285 views1 answer
Py hello code
Python
28 votes224 views2 answers
Np zeroes
Python
0 vote268 views1 answer
Find no of iterations in python
Python