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 vote172 views1 answer
Start virtualenv with python version
Python
4 votes164 views1 answer
Round python with list
Python
10 votes181 views1 answer
Hypixel main ip
Python
0 vote170 views1 answer
Printable characters python
Python
0 vote181 views1 answer
Return a new RDD containing the distinct elements in this RDD.
Python
12 votes176 views5 answers
Python random uuid
Python
41 votes178 views6 answers
Python comments
Python
0 vote264 views1 answer
List -1 python
Python
0 vote162 views1 answer
Python seaborn violin stack overflow
Python
1 vote185 views1 answer
Python hide details
Python