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

2 votes343 views1 answer
Python dictonary of dictonary
Python
0 vote392 views1 answer
Merge two Querysets in Python Django while preserving Queryset methods
Python Django
3 votes352 views2 answers
Enum
Python
62 votes339 views3 answers
Python how to loop in a list
Python
31 votes430 views1 answer
Flask vs django
Python Django
0 vote348 views1 answer
How to append panda columns using loop
Python
1 vote344 views1 answer
Time it in jupyter notebook
Python
0 vote345 views1 answer
Random pick between given things python
Python
4 votes337 views4 answers
Download pdf python
Python
0 vote334 views1 answer
Change the frequency to column in pandas
Python