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 vote128 views1 answer
Sns histplot nan values
Python
0 vote117 views1 answer
Foreign key on delete arguments
Python
0 vote138 views1 answer
Pyspark rdd sort by value descending
Python
2 votes135 views2 answers
Python snippets
Python
2 votes139 views1 answer
Slicing in pandas
Python
0 vote161 views1 answer
Django cms api create page
Python Django
0 vote174 views2 answers
Django rest framework authentication
Python Django
-1 vote174 views1 answer
Count duplicates in one column pandas
Python
6 votes178 views1 answer
How to open any program on python
Python
0 vote1.6K views1 answer
Tqdm continues afer break
Python