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 vote197 views2 answers
Why mentioning user agent in request library
Python
0 vote204 views2 answers
Python C-like structs
Python
0 vote235 views1 answer
Boxplot for all columns in python
Python
0 vote209 views1 answer
How to create empty series in pandas
Python
0 vote197 views1 answer
Django serializer,
Python
1 vote196 views2 answers
Setup VS Code for python
Python
1 vote201 views1 answer
Remove r and n from string python
Python
76 votes263 views6 answers
Select a random element from a list python
Python
0 vote348 views1 answer
Setting a conditional variable in python. Using an if else statement in python.
Python
1 vote301 views1 answer
Filter with different operator in django
Python