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

10 votes179 views3 answers
Package in python
Python
0 vote241 views2 answers
Messe graphen erstellen python
Python
1 vote173 views1 answer
Access-Control-Allow-Origin django
Python
1 vote184 views1 answer
Plt equal axis
Python
1 vote185 views1 answer
Python list add if not present
Python
0 vote179 views1 answer
Minio python remove an object
Python
0 vote180 views1 answer
Df groupby loop
Python
0 vote180 views1 answer
Python wifi moudel [WinError 2] The system cannot find the file specified
Python
2 votes222 views3 answers
Sklearn version
Python
0 vote214 views1 answer
Split into list into even chunks
Python