Breaking News: Grepper is joining You.com. Read the official announcement!

pytorch squeeze

Open Ocelot answered on October 27, 2020 Popularity 10/10 Helpfulness 3/10

answer pytorch squeeze

pytorch squeeze

Comment

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])

Popularity 10/10 Helpfulness 3/10 Language python

Source: pytorch.org

Tags: python pytorch

Link to this answer
Share Copy Link

Contributed on Oct 27 2020

Open Ocelot

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

pytorch squeeze

Contents

More Related Answers

pytorch squeeze

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.