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 vote176 views1 answer
Send arrays over POST by Python requests library
Python
1 vote130 views1 answer
Truncate date to midnight in pandas column
Python
1 vote147 views1 answer
Liczby zespolone python
Python
63 votes163 views4 answers
Python find in string
Python
0 vote118 views1 answer
Highlight max value in table pandas dataframe
Python
1 vote125 views3 answers
How to find unique values in list in python
Python
11 votes133 views6 answers
Enumerate in python
Python
16 votes227 views5 answers
Python manage.py collectstatic
Python
0 vote135 views1 answer
Df.loc jupyter
Python
2 votes122 views1 answer
Tk.destroy arguments
Python