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 vote183 views1 answer
Python - from most_similar to ldictionary
Python
2 votes186 views5 answers
Django documentation
Python
1 vote227 views1 answer
How to import PyMem python
Python
0 vote186 views1 answer
Max deviation in pandas
Python
0 vote180 views1 answer
Cvtcoloer opencv
Python
20 votes241 views4 answers
Button tkinter
Python
2 votes181 views3 answers
Pandas show column types
Python
8 votes180 views3 answers
Cool Things To Do With Python
Python
1 vote192 views1 answer
Polynomial regression using scikit-learn library
Python
1 vote202 views1 answer
Python check if number is complex
Python