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 votes314 views3 answers
E: Unable to locate package python3-pip
Python
6 votes363 views3 answers
Python print user input
Python
3 votes293 views2 answers
Python send get request with headers
Python
34 votes332 views6 answers
Beautifulsoup python
Python
1 vote285 views2 answers
Pyplot not show axis
Python
2 votes276 views1 answer
How to make an empty variable in python
Python
0 vote324 views1 answer
Date format django template filter
Python Django
0 vote284 views1 answer
Python make sound when finished
Python
1 vote284 views1 answer
Django proper capitalization case jinja
Python
0 vote318 views1 answer
Tkinter titre fenetre
Python