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

15 votes233 views4 answers
Check if a list contains any item from another list python
Python
0 vote151 views1 answer
Onehotencoder example
Python
16 votes176 views3 answers
Update tensorflow pip
Python
0 vote174 views1 answer
Equivalenci EN PYTHON DE INPUT EN C#
Python
0 vote176 views1 answer
Return a new RDD by applying a function to each element of this RDD.
Python
0 vote188 views1 answer
Python - give a name to index column
Python
0 vote173 views1 answer
Protected vs private python
Python
0 vote162 views1 answer
Python empty constructor
Python
-3 votes184 views1 answer
Plot time series python
Python
0 vote181 views1 answer
Get rid of axes numbers matplotlib
Python