Torch.unsqueeze

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

Torch.unsqueeze

a = torch.randn(4, 4, 4)
>>> torch.unsqueeze(a, 0).size()
torch.Size([1, 4, 4, 4])

>>> torch.unsqueeze(a, 1).size()
torch.Size([4, 1, 4, 4])

>>> torch.unsqueeze(a, 2).size()
torch.Size([4, 4, 1, 4])

>>> torch.unsqueeze(a, 3).size()
torch.Size([4, 4, 4, 1])

Source: linuxpip.org

Similar codes

0 vote314 views1 answer
Ascii value of pi symbol in python
Python
0 vote329 views1 answer
Tensorflow io check file exist
Python
0 vote353 views1 answer
Running code once in a while loop python
Python
4 votes318 views1 answer
What is cleaned data in django
Python
0 vote314 views2 answers
Python decimal remove trailing zero
Python
2 votes338 views6 answers
Scikit learn
Python
1 vote308 views1 answer
How to merge rows in pandas dataframe
Python
2 votes326 views2 answers
Convert np shape (a,) to (a,1)
Python
1 vote301 views1 answer
How to install drivers for selenium python
Python
3 votes305 views1 answer
Delete element from list value
Python