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 vote296 views1 answer
S = 0 for x in seq: s += x
Python
0 vote315 views1 answer
Seasonal_decompose python
Python
2 votes229 views2 answers
How to play audio in python
Python
1 vote275 views3 answers
Isprime python
Python
0 vote289 views1 answer
.lowertkinter
Python
-1 vote296 views2 answers
Mro in python
Python
22 votes353 views3 answers
How to get tomorrow date in python
Python
1 vote294 views1 answer
Python synonym library
Python
0 vote329 views1 answer
Swapping upper case and lower case string python
Python
0 vote270 views1 answer
Python double underscore methods
Python