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 vote392 views1 answer
Super title python
Python
3 votes312 views2 answers
Python get utc time
Python
34 votes334 views6 answers
How to open a file in python
Python
-1 vote315 views1 answer
Python csv writer row by row
Python
0 vote314 views1 answer
Matplotlib radial averaging
Python
0 vote339 views1 answer
Add text to plot python scatter
Python
11 votes334 views2 answers
Change pandas column value based on condition
Python
0 vote317 views1 answer
Plant python documentation
Python
0 vote322 views1 answer
Selenium chrome options suppress warnings python
Python
19 votes336 views4 answers
How to use python to sleep if the user is not using the system
Python