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 vote74 views1 answer
"_tkinter.TclError: bad entry index "1.0""
Python
3 votes124 views5 answers
See tensorflow version python
Python
0 vote122 views1 answer
Dot operator in python
Python
39 votes127 views3 answers
How to write to the end of a file in python
Python
0 vote132 views1 answer
How to hack using python
Python
32 votes124 views4 answers
Python pip install matplotlib
Python
0 vote130 views1 answer
Student notebook (finish), INB (finish), Food and Fitness log (log necessary), debate speech (finish)
Python
0 vote138 views1 answer
Numpy datetime64 get day
Python
5 votes205 views4 answers
Xticks and yticks matplotlib
Python
0 vote150 views1 answer
Pandas increment value on condition
Python