Torch.stack example

a.size()  # 2, 3, 4
b.size()  # 2, 3
b = torch.unsqueeze(b, dim=2)  # 2, 3, 1
# torch.unsqueeze(b, dim=-1) does the same thing

torch.stack([a, b], dim=2)  # 2, 3, 5

Source: newbedev.com

Torch.stack example

>>> a = torch.randn([2, 3, 4])
>>> b = torch.randn([2, 3, 4])
>>> torch.stack([a, b]).shape
torch.Size([2, 2, 3, 4])

Source: newbedev.com

Torch.stack example

>>> import torch
>>> a = torch.randn([2, 3, 4])
>>> b = torch.randn([2, 3])
>>> b = b.unsqueeze(dim=2)
>>> b.shape
torch.Size([2, 3, 1])
>>> torch.cat([a, b], dim=2).shape
torch.Size([2, 3, 5])

Source: newbedev.com

Similar codes

0 vote290 views1 answer
Calculate speed with time in datetime python
Python
5 votes299 views2 answers
Python if any element in string
Python
0 vote298 views1 answer
Diamond shape alphabatical pattern program in python
Python
0 vote304 views1 answer
Python pretty print list of tuples
Python
0 vote373 views1 answer
How to make print float value without scientific notation in dataframe in jupyter notebook
Python
0 vote346 views1 answer
Clear output alternative for visual studio code in python
Python
5 votes361 views2 answers
Selenium.common.exceptions.SessionNotCreatedException: Message: session not created: This version of ChromeDriver only supports Chrome version 81
Python
0 vote342 views1 answer
Compute confusion matrix using python
Python
3 votes313 views1 answer
Python strlen
Python
21 votes297 views5 answers
Base64 python decode
Python