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

57 votes198 views4 answers
Get date and time in python
Python
0 vote234 views2 answers
Pandas rename columns by position
Python
1 vote190 views2 answers
Rename files in a folder python
Python
18 votes186 views5 answers
Python auto clicker
Python
84 votes205 views6 answers
How to sort in pandas
Python
0 vote196 views1 answer
How to encode hexadecimal python
Python
5 votes210 views1 answer
Random forest regressor python
Python
0 vote183 views1 answer
Filter a pandas dataframe by length of list in a column
Python
9 votes256 views2 answers
Flask how to run app
Python Flask
10 votes199 views3 answers
Python click on screen
Python