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

7 votes232 views4 answers
Tensorflow gpu test
Python
0 vote235 views1 answer
How to select name parent table in model laravel
Python
10 votes224 views4 answers
Create empty numpy array
Python
1 vote264 views1 answer
Django loop index
Python
8 votes230 views3 answers
Get_dummies
Python
0 vote216 views1 answer
Pandas iloc include header
Python
0 vote245 views1 answer
Gun in python turtle
Python
0 vote228 views1 answer
Pandas groupby without reset index
Python
0 vote242 views1 answer
Py convert binary to int
Python
3 votes242 views3 answers
Spacex
Python