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 vote133 views1 answer
Django auto increment field
Python Django
0 vote100 views1 answer
Separating tuple in pandas
Python
1 vote98 views1 answer
Replace url with text python
Python
77 votes99 views6 answers
Reverse a string python
Python
8 votes107 views2 answers
How to earse special chrat¥cter from string in python
Python
2 votes127 views6 answers
E: Could not get lock /var/lib/dpkg/lock-frontend - open (11: Resource temporarily unavailable)
Python
0 vote101 views1 answer
Python save base64 temp file
Python
0 vote107 views1 answer
Add key if not exists python
Python
0 vote122 views1 answer
Keep tkinter window below others
Python
4 votes97 views2 answers
Replace nan using fillna
Python