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

-1 vote169 views1 answer
Separate digits with comma
Python
0 vote171 views2 answers
How to decrease size of graph in plt.scatter
Python
0 vote165 views1 answer
Grapper
Python
0 vote208 views1 answer
Split list in 3 part
Python
0 vote166 views1 answer
Python multiple items in with statment
Python
0 vote169 views1 answer
Plot every nth label in barplot
Python
0 vote165 views1 answer
Python dataframe add rank column
Python
11 votes206 views6 answers
Pandas join dataframe
Python
4 votes171 views1 answer
Pandas check if value in column is in a list
Python
3 votes162 views1 answer
Python cmd to remove file
Python