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 vote252 views1 answer
Vertica long running queries
Python
5 votes261 views3 answers
How to run flask in port 80
Python
2 votes195 views2 answers
Horizontal bar plot python
Python
1 vote184 views1 answer
Url in form action django
Python
0 vote2.1K views1 answer
How to print cow on python
Python
88 votes179 views6 answers
Save dataframe to csv
Python
1 vote192 views1 answer
Rum system commands python
Python
0 vote194 views1 answer
How to count number from 1 to 10 in python
Python
14 votes259 views6 answers
Binary search in python
Python
1 vote191 views1 answer
Mean Kurtosis of all rows pandas
Python