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

2 votes240 views3 answers
Rotate matrix 90 degrees clockwise python
Python
2 votes233 views2 answers
Create pdf from images python
Python
-1 vote241 views2 answers
Scrapy shell
Python
4 votes214 views1 answer
Dropdown in tkinter
Python
0 vote231 views1 answer
Check how many days old file is python
Python
0 vote233 views1 answer
What is horse riding sport name
Python
1 vote223 views1 answer
Read bytes from file python
Python
1 vote211 views1 answer
Check string equal with regular expression python
Python
45 votes227 views6 answers
Python webbrowser
Python
0 vote233 views1 answer
How to use drf pagination directly
Python