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 votes137 views2 answers
How to earse special chrat¥cter from string in python
Python
0 vote136 views1 answer
Python set match two list
Python
0 vote129 views1 answer
Come mettere una scelta su python
Python
2 votes238 views1 answer
Django and react url conflict
Python Django
0 vote144 views1 answer
Whitespace delimiter python
Python
4 votes138 views1 answer
Convert string to lowercase in python
Python
13 votes132 views3 answers
How to split text into list python by characters
Python
0 vote134 views1 answer
TypeError: attrib() got an unexpected keyword argument 'convert'
Python
0 vote148 views1 answer
How to make python print 2 line text in one code
Python
35 votes127 views6 answers
Create pandas dataframe from dictionary
Python