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 vote134 views2 answers
Check status code urllib
Python
10 votes122 views6 answers
Add to python list
Python
0 vote131 views1 answer
Python - Cómo Jugar archivo Mp3
Python
0 vote250 views3 answers
Django prevent duplicate entries
Python Django
0 vote116 views1 answer
Python openpyxl csv to excel
Python
38 votes128 views3 answers
Python print datetime
Python
0 vote151 views1 answer
Code=H18 desc="Server Request Interrupted" django
Python
0 vote145 views1 answer
Pandas df to mongodb
Python
2 votes147 views2 answers
Import other python files
Python
1 vote188 views1 answer
Open a filename starting with in python
Python