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 vote132 views1 answer
Hsv to rgb python
Python
0 vote198 views1 answer
Tadjust margines automatically matplotlib
Python
3 votes155 views1 answer
Drop duplicates pandas first column
Python
3 votes135 views3 answers
Python pip not working
Python
4 votes140 views2 answers
Python map input
Python
0 vote145 views1 answer
Saree
Python
7 votes139 views2 answers
What is the output of the following code? np.arange(0,22,6)
Python
0 vote164 views1 answer
Colorbar min max matplotlib
Python
0 vote136 views1 answer
Discord py server.channels
Python
9 votes150 views5 answers
Pandas to dictionary
Python