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

40 votes178 views5 answers
New line print python
Python
2 votes133 views1 answer
.env files in repl
Python
0 vote134 views1 answer
Python is space
Python
0 vote160 views1 answer
Case insensitive replace python
Python
0 vote129 views1 answer
Invalid Python Environment: Python is unable to find Maya's Python modules
Python
1 vote120 views1 answer
Extract minutes from timedelta python
Python
0 vote133 views1 answer
Save artist animation puython
Python
0 vote149 views1 answer
Add a value to the start of a list python
Python
1 vote137 views1 answer
Write a python program to find gcd of two numbers
Python
0 vote136 views1 answer
Flask sqlalchemy query specific columns
Python