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

1 vote180 views3 answers
Management commands django
Python
0 vote170 views6 answers
Gdal user with anaconda
Python
8 votes191 views3 answers
Convert dataframe to json
Python
183 votes197 views6 answers
How to update pip python
Python
4 votes165 views1 answer
What makes the bot talk? is there a word i have to say?
Python
0 vote264 views2 answers
Plt.xlabel not working
Python
13 votes170 views2 answers
Min int python
Python
0 vote179 views1 answer
Free download django app for windows 10
Python
0 vote176 views1 answer
List files in http directory python
Python
3 votes181 views4 answers
Pair Plot Seaborn
Python