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 vote143 views1 answer
Decode in django templates
Python
2 votes155 views10 answers
@7chats
Python
4 votes190 views1 answer
Sort dict by value
Python
3 votes158 views1 answer
How to connect templates in django
Python
0 vote158 views1 answer
Pandas dataframe from list how to make the date column an index
Python
-2 votes156 views1 answer
How to check if a string contains spaces in python
Python
2 votes170 views1 answer
Django.core.exceptions.ImproperlyConfigured
Python
0 vote165 views2 answers
Python convert string to list of dictionaries
Python
3 votes200 views2 answers
How to create a label in python
Python
21 votes166 views6 answers
Select columns from dataframe pandas
Python