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

1 vote359 views2 answers
Python write json to file utf8
Python
0 vote329 views1 answer
Add silence to the end of an MP3 python
Python
0 vote334 views1 answer
How to list gym envirolments
Python
1 vote361 views1 answer
Matplotlib ax.annotate color of the arrow
Python
11 votes392 views3 answers
Python print list with newline
Python
0 vote313 views10 answers
Python open file partially
Python
14 votes314 views6 answers
Permutations in python
Python
1 vote332 views1 answer
Python trick big numbers visualisation
Python
0 vote334 views1 answer
Post to instagram from pc python
Python
0 vote361 views1 answer
Sum
Python