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 vote191 views1 answer
What is proc file
Python
15 votes180 views4 answers
Make a condition statement on column pandas
Python
0 vote196 views1 answer
Find root of the path of file os package
Python
2 votes187 views2 answers
Generate random integer matrix python
Python
86 votes206 views3 answers
How to read json from python
Python
1 vote202 views2 answers
How to get elasticsearch index list using python
Python
2 votes184 views3 answers
0 1 1 2 3 5 in python
Python
5 votes199 views2 answers
Django password field
Python
8 votes196 views6 answers
Remove outliers python pandas
Python
30 votes196 views5 answers
Generate random float python
Python