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 vote327 views1 answer
Imagedatagenerator split train test
Python
0 vote320 views1 answer
Tuple parameter function python is None
Python
2 votes349 views1 answer
How to compare two text files in python
Python
0 vote292 views1 answer
If we use list in the dictionary
Python
0 vote340 views1 answer
Python set negative infinity
Python
0 vote312 views1 answer
Workbook select sheet python
Python
13 votes323 views6 answers
Range() python
Python
29 votes325 views2 answers
Python string upper method
Python
5 votes317 views1 answer
Generate a color python
Python
0 vote345 views1 answer
Send command civil3D
Python