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 vote277 views1 answer
Extract bigrams python
Python
2 votes276 views1 answer
Pygityb
Python
2 votes251 views1 answer
Favicon django
Python
0 vote285 views1 answer
Split pandas dataframe in two
Python
0 vote258 views1 answer
Python print os command output -1
Python
1 vote304 views1 answer
Flask socketio usage
Python Flask
0 vote265 views1 answer
Pandas dataframe select last n columns
Python
2 votes267 views1 answer
Pass variable in subprocess run python
Python
1 vote260 views1 answer
Pandas add character to string
Python
5 votes319 views1 answer
Flask boiler plate
Python Flask