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 vote118 views1 answer
Rotate image python
Python
40 votes115 views6 answers
Pandas replace null with 0
Python
36 votes108 views2 answers
Python file open modes
Python
8 votes102 views3 answers
Linear search python
Python
11 votes119 views5 answers
Capitalize first letter in python
Python
13 votes124 views6 answers
Sort list python
Python
1 vote143 views2 answers
How explode by using two columns pandas
Python
1 vote107 views2 answers
Number of columns with no missing values
Python
0 vote107 views1 answer
Reddit python 3 time a python program
Python
92 votes75 views3 answers
Python self
Python