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 vote209 views1 answer
Python os check if file with extension exists
Python
0 vote191 views1 answer
Colorbar remove tick lines and border
Python
0 vote202 views1 answer
Django update request.post
Python Django
10 votes182 views3 answers
Count occurrence in array python
Python
1 vote194 views3 answers
Sum method in python
Python
0 vote182 views1 answer
Python mixins
Python
0 vote197 views1 answer
Pandas series remove element by index
Python
8 votes189 views5 answers
What is module in python
Python
0 vote194 views1 answer
Qt designer messagebox python
Python
2 votes176 views1 answer
Is pythin a real coding language
Python