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 vote158 views1 answer
Pandas df.index.values
Python
7 votes156 views4 answers
Pandas filter rows by value
Python
0 vote278 views1 answer
Python catch output to terminal
Python
6 votes158 views2 answers
Python sort the values in a dictionary
Python
1 vote164 views2 answers
Winsound python
Python
0 vote186 views1 answer
Read a csv and plot in python
Python
0 vote157 views1 answer
Concatenar columnas en una del mismo dataset
Python
0 vote212 views1 answer
Using rstrip() method to remove the newline character from a string
Python
2 votes161 views2 answers
Pandas divide one column by another
Python
1 vote144 views1 answer
Get content of one column in pandas
Python