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

16 votes256 views1 answer
Super idol
Python
0 vote269 views1 answer
12
Python
40 votes265 views4 answers
Find todays date in python
Python
0 vote279 views1 answer
Return an RDD of grouped items.
Python
2 votes267 views1 answer
How to make rich presence discord,py
Python
0 vote272 views1 answer
Remove prefix from string python
Python
0 vote261 views4 answers
Python call function from string
Python
0 vote282 views1 answer
For loop to create a set of counters in python
Python
13 votes279 views2 answers
Only keep rows of a dataframe based on a column value
Python
0 vote268 views2 answers
Pandas series to numpy array
Python