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

0 vote229 views1 answer
Python empty constructor
Python
43 votes226 views5 answers
Pandas iterrows
Python
0 vote257 views1 answer
Set list start at 1 python
Python
0 vote245 views1 answer
Pandas ta quick start example
Python
0 vote277 views1 answer
How to select the shortest item in a python list
Python
1 vote236 views3 answers
How to access variable of one function in another function in python
Python
8 votes252 views1 answer
Iterative binary search python
Python
1 vote279 views2 answers
Install fasttext python
Python
1 vote256 views1 answer
Create square matrix python
Python
0 vote249 views1 answer
Python alfabet
Python