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 vote186 views1 answer
Read api from django
Python
5 votes198 views2 answers
Exemple python gradient
Python
42 votes199 views3 answers
Df iterrows pandas
Python
0 vote197 views1 answer
Affinity propagation python
Python
0 vote184 views1 answer
Get midnight of current day python
Python
1 vote175 views1 answer
Python requests.get pdf An appropriate representation of the requested resource could not be found
Python
-1 vote182 views1 answer
Usage code grepper
Python
0 vote188 views1 answer
Group by, aggregate multiple column -pandas
Python
0 vote184 views1 answer
Python import module from bitbucket
Python
0 vote183 views1 answer
Redirect the Python Script Output to File
Python