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 vote137 views1 answer
Cannot import name
Python
0 vote184 views3 answers
Python show png
Python
0 vote212 views1 answer
Python except keyboardinterrupt
Python
71 votes175 views5 answers
How to make loop python
Python
130 votes154 views10 answers
Conda create environment
Python
0 vote189 views1 answer
Blockchain private key generator
Python
1 vote168 views1 answer
Resto division python
Python
0 vote224 views1 answer
Django builtin signals
Python Django
0 vote236 views1 answer
Get span text selenium python
Python
1 vote184 views1 answer
How to provide default value when assign i ngvariables python
Python