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

9 votes150 views2 answers
Matplotlib axis rotate xticks
Python
0 vote144 views1 answer
Pandas csv sum column
Python
0 vote155 views1 answer
Cv2.videocapture python set frame rate
Python
9 votes157 views6 answers
How to print something in python
Python
1 vote154 views1 answer
How do i check if a django queryset is empty
Python
1 vote178 views2 answers
How to get the number of rows and columns in a numpy array
Python
0 vote144 views2 answers
How to give order in boxplot matplotlib
Python
0 vote152 views1 answer
This branch is for urllib3 v1.22 and later
Python
2 votes151 views1 answer
Save file python tkinter
Python
0 vote149 views1 answer
Pyqt button clicked connect
Python