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 vote307 views1 answer
Python string contains substring ignore case
Python
0 vote256 views1 answer
Dataframe how to find columns that start with prefix
Python
11 votes256 views4 answers
Python get random character from string
Python
0 vote249 views1 answer
Python advanced programming exercises
Python
5 votes278 views2 answers
Python 3 replace all whitespace characters
Python
0 vote260 views2 answers
Compiling python code
Python
0 vote253 views2 answers
Pdf to word
Python
2 votes262 views2 answers
How to find anyone's ip address
Python
8 votes254 views3 answers
Django sort queryset
Python
4 votes267 views2 answers
Django template for range
Python