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 vote124 views1 answer
Python find oldest and newest date
Python
0 vote126 views1 answer
How to use with statement in python 2.5 and earlier
Python
12 votes145 views2 answers
Print json python
Python
24 votes137 views6 answers
Remove last element from list python
Python
65 votes118 views6 answers
Spanish dictionary
Python
0 vote120 views1 answer
How to create api for get data in django rest framework
Python
1 vote143 views1 answer
Numpy sign method
Python
0 vote125 views1 answer
Max fonction code in python
Python
142 votes141 views6 answers
While loop python
Python
1 vote96 views1 answer
Jupyter clear output
Python