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 vote286 views1 answer
How to change size of xticks
Python
1 vote270 views1 answer
Deleting an object in python
Python
9 votes272 views2 answers
Python delete all data from file
Python
0 vote277 views1 answer
Documentation on fasttext gensim python
Python
0 vote303 views1 answer
Python depth first search
Python
0 vote276 views1 answer
Read page source from text file python
Python
1 vote261 views2 answers
Python pd.DataFrame.from_records remove header
Python
7 votes284 views5 answers
Python csv to list
Python
0 vote277 views1 answer
Comparison python
Python
0 vote297 views1 answer
Statistique in python
Python