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

4 votes255 views4 answers
How to shuffle array in python
Python
1 vote250 views1 answer
Pylint: disable=unused-argument
Python
0 vote244 views1 answer
Limit of list in python
Python
1 vote418 views1 answer
Backup django db from one database to another
Python Django
3 votes252 views1 answer
Json not readable python
Python
0 vote244 views1 answer
Python args description multiple lines
Python
0 vote258 views1 answer
Bs4 table examples python
Python
2 votes259 views3 answers
List inside a list in python
Python
3 votes258 views2 answers
How to make a forever loop in python
Python
0 vote256 views1 answer
List Get a Element-2
Python