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

0 vote330 views1 answer
Numpy distance of consecutive elements
Python
0 vote347 views1 answer
Scaling pkl file?
Python
1 vote342 views1 answer
Pandas forward fill after upsampling
Python
1 vote404 views2 answers
Changing the number of ticks on a Matplotlib plot axis
Python
0 vote342 views1 answer
Show post id on django admin interface
Python
39 votes343 views6 answers
How to use a class in python
Python
0 vote337 views1 answer
Check python version conda env
Python
0 vote326 views1 answer
Dataframe time index convert tz naive to tz aware
Python
39 votes317 views6 answers
Python float precision
Python
6 votes348 views4 answers
Openpyxl
Python