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

-1 vote174 views1 answer
Python data first column indices
Python
30 votes178 views2 answers
Matplotlib change text size
Python
7 votes168 views2 answers
How to convert uppercase to lowercase and vice versa in python
Python
0 vote228 views2 answers
Numpy softmax
Python
0 vote178 views1 answer
Python to apk converter software download for pc
Python
0 vote186 views1 answer
Create new column with length of old column value python
Python
9 votes190 views4 answers
Check os python
Python
32 votes175 views3 answers
Python move files
Python
11 votes181 views6 answers
Python list
Python
14 votes191 views6 answers
If condition python
Python