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

8 votes214 views2 answers
Pandas iterate columns
Python
1 vote194 views1 answer
Required validator python WTForms
Python
0 vote214 views1 answer
Python remove dtype from array
Python
0 vote197 views1 answer
Python socket set title
Python
0 vote204 views1 answer
When converting from dataframe to list delete nan values
Python
0 vote202 views1 answer
Np.argmax python could not be evaluated
Python
0 vote226 views1 answer
Python string: string concatenation
Python
1 vote220 views1 answer
Test Speed internet using Python
Python
1 vote199 views1 answer
Find absolut vale in python
Python
2 votes213 views2 answers
Php echo shorthand
Python