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 vote130 views1 answer
Production mode flask
Python
1 vote129 views1 answer
Merge multiple csv files
Python
80 votes145 views6 answers
Sort dictionary by value python
Python
91 votes172 views6 answers
Python for in
Python
1 vote129 views1 answer
Filter dict
Python
41 votes142 views4 answers
Find out current datetime in python
Python
12 votes132 views1 answer
Python regex cheat sheet
Python
2 votes122 views3 answers
Solve equation python
Python
0 vote138 views1 answer
Withdraw() opposite tjinter
Python
0 vote148 views1 answer
Sort first element reverse sort second python
Python