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

4 votes98 views4 answers
Distribution plot python
Python
-1 vote144 views1 answer
Insert into string python
Python
4 votes110 views2 answers
How to concat on the basis of particular columns in pandas
Python
0 vote102 views1 answer
How to read docx file in python
Python
5 votes104 views1 answer
Make jupyter notebook wider
Python
5 votes111 views6 answers
Delete a file from git history
Python
9 votes111 views1 answer
Create models in django
Python
16 votes124 views2 answers
Python reload class
Python
3 votes101 views1 answer
From collections import defaultdict
Python
9 votes118 views1 answer
Delete model object django
Python Django