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

10 votes105 views3 answers
Cv show image python
Python
0 vote118 views1 answer
Python background process
Python
-1 vote106 views2 answers
Crop image python
Python
0 vote125 views1 answer
Docker opencv python libGL.so.1: cannot open shared object file: No such file or directory
Python
14 votes117 views2 answers
Python check if dictionary empty
Python
0 vote105 views1 answer
Pygame tick time
Python
0 vote141 views1 answer
Access host database django docker
Python Django
0 vote112 views1 answer
Http.client.invalidurl: url can't
Python
1 vote114 views1 answer
If a number times a number is true python
Python
0 vote109 views1 answer
How to leave a function python
Python