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

6 votes152 views4 answers
How to open folder in python
Python
3 votes189 views2 answers
Reciprocal python
Python
1 vote153 views1 answer
Why is python so popular
Python
24 votes153 views3 answers
Flask vs django
Python
-6 votes151 views1 answer
Download any file from link colab
Python
1 vote162 views2 answers
Download pdf from url python
Python
-1 vote146 views1 answer
Python graph
Python
0 vote149 views1 answer
Sublim text python indentation error
Python
1 vote182 views1 answer
Python access modifiers
Python
0 vote112 views1 answer
Nrf24l01 arduino raspberry pi
Python