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

1 vote104 views1 answer
How to loc all cells except python
Python
0 vote93 views1 answer
Multivariable traces f(x, y) = sin(x)cos(y)
Python
0 vote118 views1 answer
Python regex find single character
Python
0 vote134 views1 answer
Python regular expression escape vertical bar |
Python
7 votes130 views2 answers
How to code python
Python
4 votes110 views3 answers
How to remove numbers from string in python dataframe
Python
0 vote100 views1 answer
Pandas describe get mean min max
Python
0 vote114 views1 answer
Raspberry pi run python script on boot
Python
0 vote144 views1 answer
Pandas count empty string values
Python
3 votes109 views1 answer
Load saved model tensorflow
Python