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 vote106 views1 answer
How to load wav file with python
Python
0 vote96 views1 answer
Write page source to text file python
Python
93 votes97 views6 answers
Creating virtual environment python
Python
0 vote125 views1 answer
Repetition of word in python
Python
15 votes103 views1 answer
Pandas merge python
Python
0 vote131 views1 answer
Lipa na mpesa daraja python
Python
0 vote90 views1 answer
Python find duplicates in string
Python
20 votes92 views5 answers
Remove first character from string python
Python
0 vote120 views1 answer
How to reverse a list in python without using inbuilt function
Python
42 votes115 views3 answers
Python get command line arguments
Python