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 vote143 views2 answers
Shotgun api filter
Python
0 vote167 views1 answer
How to preserve white space when joining an array python
Python
44 votes137 views3 answers
Find sum numbers in a list in python
Python
5 votes153 views2 answers
Df.shape 0
Python
0 vote105 views1 answer
Pandas read all csv in folder
Python
0 vote166 views1 answer
Django 3 create async rest api
Python
0 vote148 views1 answer
Who is agada nathan
Python
0 vote133 views1 answer
Numpy index array all except
Python
9 votes163 views6 answers
Django filters
Python
0 vote137 views1 answer
Check if argv exists python
Python