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 vote331 views1 answer
Write a python program which accepts the user
Python
71 votes313 views5 answers
How to make loop python
Python
35 votes326 views6 answers
Install matplotlib
Python
-1 vote343 views3 answers
Transpose of a matrix in python numpy
Python
0 vote360 views1 answer
Tessa thompson
Python
12 votes343 views1 answer
Python range for float
Python
8 votes327 views2 answers
Import Image
Python
0 vote328 views1 answer
Slack bot error not_in_channel
Python
0 vote328 views1 answer
Free download django app for windows 10
Python
0 vote385 views1 answer
Matplotlib bar chart value_counts
Python