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 vote143 views1 answer
Localhost postgres settings django
Python Django
1 vote186 views1 answer
Pygame disable message
Python
1 vote161 views1 answer
Height width image opencv
Python
0 vote178 views1 answer
Split the string using the separator
Python
1 vote174 views3 answers
Making ckeditor django responsive
Python
0 vote167 views1 answer
Write dict to json file with special characters
Python
4 votes169 views1 answer
Python windows hide files
Python
0 vote173 views1 answer
Switch case python 3.10
Python
0 vote171 views1 answer
Np.argmax python could not be evaluated
Python
0 vote189 views2 answers
Create a list of pandas index
Python