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 vote225 views1 answer
Calculate highest frequency or mode in pandas dataframe
Python
0 vote268 views1 answer
How to update sklearn using conda
Python
1 vote232 views2 answers
ModuleNotFoundError: No module named 'requests_toolbelt'
Python
0 vote184 views1 answer
Which clause is used to place condition with GROUP BY clause in a table
Python
0 vote196 views1 answer
Return more than one value python
Python
-1 vote203 views1 answer
Ipywidgets pip
Python
0 vote275 views1 answer
How to load keras model from json
Python
0 vote183 views1 answer
Flask migrate multiple heads
Python
24 votes177 views2 answers
Convert list of int to string python
Python
1 vote198 views1 answer
Read csv python without id
Python