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 vote183 views2 answers
Remove rows if not matching with value in df
Python
0 vote132 views1 answer
Calling function whose name is in a variable
Python
0 vote184 views1 answer
How to get column names having numeric value in pandas
Python
0 vote150 views1 answer
Python get the name of the song that is playing
Python
1 vote140 views2 answers
Right angle triangle in python
Python
2 votes128 views1 answer
Not scientific notation python
Python
0 vote136 views1 answer
Get most recurring element in a list python
Python
0 vote159 views1 answer
Identify null values
Python
0 vote133 views3 answers
How to draw dendrogram in python
Python
5 votes175 views4 answers
Import user in django
Python Django