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 vote272 views1 answer
> 'depth' is 6 (CV_64F)
Python
0 vote254 views1 answer
PyPi module to make your terminal look like hacking
Python
0 vote274 views1 answer
Shutdown flask server with request
Python
9 votes326 views2 answers
Flask flash
Python Flask
-4 votes305 views1 answer
Hello world python 2
Python
0 vote264 views1 answer
Arrange array in ascending order python
Python
0 vote255 views1 answer
Cget subassembly civid3d
Python
0 vote253 views1 answer
Share x axis matplotlib
Python
0 vote252 views2 answers
Convert pdf folder to excell pandas
Python
0 vote259 views2 answers
How to make a bill in python
Python