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

5 votes193 views2 answers
Python all lowercase letters
Python
1 vote159 views1 answer
Debconf: falling back to frontend: Readline Configuring tzdata
Python
1 vote188 views1 answer
How to remove trackback on python when ctrl c
Python
135 votes179 views6 answers
Replace string
Python
3 votes155 views1 answer
Local testing server Python
Python
0 vote168 views1 answer
Python create sqlite db in memory
Python
0 vote280 views1 answer
Json indent options python
Python
100 votes154 views5 answers
Python -m pip install --upgrade pip
Python
0 vote158 views1 answer
How to generate python code
Python
0 vote166 views1 answer
Converting (YYYY-MM-DD-HH:MM:SS) date time
Python