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

5 votes332 views2 answers
Hide console in python build
Python
1 vote313 views1 answer
Print value of array python
Python
0 vote319 views1 answer
Read file from form input type filepython
Python
0 vote299 views1 answer
Floating point python
Python
0 vote336 views1 answer
Python print to string
Python
0 vote309 views1 answer
Imread real color cv2
Python
0 vote314 views1 answer
Name decorator in python
Python
0 vote306 views1 answer
Python find first char index from a string stackoverflow
Python
34 votes292 views3 answers
Python string replace variable
Python
7 votes303 views1 answer
Emoji In Python
Python