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 vote312 views2 answers
Sep and end in print python
Python
0 vote295 views1 answer
Duplicate a list with for loop in python
Python
0 vote296 views1 answer
Python developer job description
Python
0 vote421 views1 answer
Unzipping the value using zip() python
Python
0 vote292 views1 answer
Pyfiglet not coming up
Python
2 votes342 views1 answer
Ln in python
Python
35 votes320 views5 answers
How to use replace in python
Python
9 votes290 views2 answers
How to make a resizable python tkinter window have a limit
Python
0 vote333 views1 answer
How to compare values in dictionary with same key python
Python
55 votes313 views6 answers
Train test split
Python