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 vote128 views1 answer
Django router multiple pk
Python
0 vote164 views1 answer
How to discover which index labels are in other
Python
2 votes150 views1 answer
Python remove empty folders
Python
0 vote135 views1 answer
Pss signatures python
Python
46 votes178 views4 answers
How to stop container in docker
Python
0 vote141 views1 answer
Build the union of a list of RDDs
Python
1 vote163 views1 answer
Python access modifiers
Python
1 vote144 views1 answer
Img read
Python
1 vote185 views1 answer
Django get form id from request
Python Django
1 vote137 views1 answer
Best Python Free Tutorial
Python