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 vote112 views2 answers
Print hexadecimal in python
Python
1 vote102 views1 answer
Python sort a list by a custom order
Python
0 vote101 views1 answer
Calculate the R^2 for X and Y python
Python
0 vote95 views1 answer
Take first 10 row while reading csv python
Python
0 vote108 views1 answer
Easy frequency analysis python
Python
0 vote121 views1 answer
Rom pathlib import Path # Build paths inside the project like this: BASE_DIR / 'subdir'. BASE_DIR = Path(__file__).resolve().parent.parent
Python
0 vote99 views1 answer
Expected a list of items but got type int . django
Python
0 vote109 views1 answer
Can't use "pyimage2" as iconphoto
Python
0 vote105 views1 answer
How to implement nfa in python
Python
0 vote115 views1 answer
How to get data after last slash in python
Python