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

-2 votes206 views2 answers
Python list with several same values
Python
0 vote214 views3 answers
Sum range
Python
1 vote183 views1 answer
Http client post python
Python
0 vote212 views1 answer
Invalid Python Environment: Python is unable to find Maya's Python modules
Python
0 vote190 views1 answer
Pandas redondear un valor
Python
0 vote168 views1 answer
Plt tickpad
Python
0 vote170 views1 answer
CharAt in Python
Python
0 vote177 views1 answer
Ubuntu 20.10 python 3.8
Python
-4 votes174 views5 answers
Python 3.5 release date
Python
1 vote214 views1 answer
How to play sound after pressing a button in tkinter
Python