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

1 vote222 views1 answer
Change the user agent selenium
Python
2 votes228 views2 answers
Python socket get client ip address
Python
3 votes207 views1 answer
Module 'cv2' has no 'videocapture' member python
Python
6 votes236 views1 answer
Python logo png
Python
0 vote233 views1 answer
Torch.unsqueze
Python
3 votes200 views2 answers
Requests post with headers python
Python
0 vote201 views1 answer
Send arrays over POST by Python requests library
Python
1 vote238 views1 answer
Python lock using a file
Python
14 votes234 views3 answers
Python tkinter define window size
Python
0 vote281 views1 answer
Circular import error
Python