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 vote181 views1 answer
Python string: .find()
Python
17 votes149 views5 answers
Find all occurrences of an element in a list python
Python
1 vote227 views5 answers
How to find avrage python
Python
1 vote169 views1 answer
Discord.py say something
Python
0 vote157 views1 answer
Bytestring python
Python
1 vote157 views1 answer
Find min and max from dataframe column
Python
0 vote164 views1 answer
Pygame for loop for draw shape
Python
14 votes175 views6 answers
Installing django on windows
Python Django
7 votes166 views2 answers
Python deepcopy
Python
0 vote171 views1 answer
Check if any entries arte none
Python