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

3 votes109 views1 answer
Python list of random float numbers
Python
0 vote108 views1 answer
Pygame.set_volume(2.0) max volume
Python
3 votes125 views2 answers
How to connect python to html
Python
54 votes111 views3 answers
Getting size of list in python
Python
0 vote126 views1 answer
Create new column with length of old column value python
Python
0 vote111 views1 answer
Norm 2 or ocklidos of matrix in python
Python
0 vote112 views1 answer
How to python program
Python
0 vote114 views2 answers
How to close a flask web server with python
Python
0 vote110 views1 answer
@foreach 1 numper
Python
2 votes195 views2 answers
Install wxPython
Python