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 vote120 views1 answer
Python get pc runtime
Python
0 vote120 views1 answer
How to import something in python
Python
0 vote139 views1 answer
Check this id exist in database django
Python Django
0 vote117 views1 answer
Python copy file create intermediate directories
Python
0 vote152 views1 answer
How to send file in django response
Python Django
0 vote108 views1 answer
Remove initial space python
Python
17 votes118 views5 answers
ModuleNotFoundError: No module named 'tkinter'
Python
3 votes123 views2 answers
Locale library python 3
Python
17 votes109 views5 answers
Python input integer
Python
0 vote70 views1 answer
Nrf24l01 arduino raspberry pi
Python