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

14 votes201 views4 answers
Pip install virtualenv windows
Python
0 vote203 views1 answer
Import sys execute cmd
Python
0 vote293 views1 answer
Oserror: invalid cross-device link
Python
0 vote199 views1 answer
Python read xls
Python
0 vote197 views1 answer
Convert 2d string array to float python
Python
0 vote196 views1 answer
ModuleNotFoundError: No module named 'timeago'
Python
34 votes240 views6 answers
Beautifulsoup python
Python
23 votes186 views2 answers
Django rest
Python
1 vote189 views1 answer
Function as parameter tpye hinting python
Python
1 vote147 views1 answer
Plus months onto python date
Python