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 votes211 views2 answers
Os.walk python
Python
0 vote216 views1 answer
Check string in a list for substrings and return index
Python
0 vote261 views3 answers
Pydrive download by url
Python
31 votes279 views6 answers
Python strip whitespace
Python
4 votes235 views2 answers
Django group by
Python
0 vote211 views1 answer
How to make python open a link
Python
0 vote262 views1 answer
Register template tag django
Python Django
8 votes252 views3 answers
How to delete nan values in python
Python
7 votes218 views1 answer
How to check if an application is open in python
Python
0 vote250 views1 answer
Count total number of null, isna sum python
Python