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

1 vote160 views1 answer
Export PyTorch model in the ONNX Runtime format
Python
0 vote157 views1 answer
Close python window after execution
Python
4 votes236 views1 answer
Finding a substring between 2 strings
Python
3 votes192 views1 answer
Load pandas from text
Python
2 votes170 views1 answer
How to set chrome options python selenium for a folder
Python
0 vote191 views1 answer
How to run mac terminal from python script
Python
1 vote156 views1 answer
Plt normalized histogram
Python
0 vote166 views1 answer
Change increment in for loop python
Python
0 vote160 views1 answer
Sha 256 python
Python
0 vote224 views1 answer
Purpose of migration folder in django
Python Django