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

2 votes317 views1 answer
Pass in queryset as filter django
Python Django
27 votes271 views2 answers
What value do we get from NULL database python
Python
0 vote268 views1 answer
Holding a function to the code in python
Python
0 vote261 views1 answer
Python set prcess name
Python
0 vote274 views1 answer
How to change scenes in python
Python
0 vote269 views1 answer
Fibonacci in list python
Python
40 votes286 views6 answers
Count nan pandas
Python
1 vote258 views1 answer
Code for python shell 3.8.5 games
Python
0 vote289 views1 answer
How to show line chart in seaborn lib
Python
5 votes262 views1 answer
Getpass
Python