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

7 votes208 views2 answers
Show image in python
Python
3 votes159 views2 answers
How to change value of categorical variable in python
Python
0 vote120 views1 answer
Keras.utils.plot_model keeps telling me to install pydot and graphviz
Python
0 vote162 views1 answer
Google translator api pyhton
Python
1 vote189 views1 answer
Cannot import name 'candlestick2_ohlc
Python
0 vote180 views1 answer
Comprehensions
Python
6 votes140 views1 answer
Convert txt file to list python
Python
0 vote166 views1 answer
Python assert is not null
Python
0 vote139 views6 answers
Python math
Python
0 vote238 views1 answer
How to set default user group in django
Python Django