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 vote142 views1 answer
Python merge csv files in same folder
Python
0 vote157 views1 answer
Select values for row matching condition
Python
0 vote144 views1 answer
Json database python
Python
0 vote143 views1 answer
Django admin manytomanyfield filter
Python Django
0 vote136 views1 answer
How to set time limit for receiving data in socket python
Python
2 votes151 views1 answer
Python check if string is in input
Python
0 vote172 views1 answer
Prefetched_related django rest framework
Python Django
6 votes147 views4 answers
Print all elements in list python
Python
0 vote134 views1 answer
Ios iterate through dictionary
Python
0 vote179 views1 answer
Error 401 unauthorized "Authentication credentials were not provided."
Python