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

1 vote311 views1 answer
Django model query add annotation field to show duplicate count
Python
0 vote314 views1 answer
Delete nans in df python
Python
1 vote314 views4 answers
Driver.find_element_by_xpath
Python
1 vote320 views1 answer
How to make a countdown in pygame
Python
0 vote320 views1 answer
How to resize image with pillow in django
Python
2 votes330 views2 answers
Remove steam from ubuntu
Python
2 votes333 views1 answer
How store list in django session
Python
0 vote380 views1 answer
Check if key exists in sesson python flask
Python Flask
0 vote305 views1 answer
AttributeError: 'Engine' object has no attribute 'runandwait' python
Python
-1 vote346 views1 answer
Let’s add a docstring to the greeting method. How about, “Outputs a message with the name of the person”.
Python