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 votes150 views1 answer
Python two string equal
Python
8 votes146 views4 answers
Remove nan from list python
Python
0 vote134 views2 answers
Remove repetitive characters from the specified column of a given DataFrame
Python
0 vote170 views1 answer
Integers come in all sizes solution in python3
Python
10 votes128 views2 answers
Python get absolute path of file
Python
0 vote140 views1 answer
Print all objects in list python
Python
-1 vote173 views1 answer
Filter groupby pandas
Python
0 vote90 views1 answer
Pip install --upgrade pip no module named pip
Python
1 vote148 views1 answer
How to calculate average in list python by using whil loop
Python
-3 votes134 views2 answers
Mean
Python