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 vote234 views1 answer
Pyqt5 image
Python
0 vote244 views1 answer
Qlcdnumber set value python
Python
5 votes199 views4 answers
Fahrenheit to celsius python program
Python
2 votes233 views1 answer
Convert opencv image to pil image
Python
0 vote232 views1 answer
The following code shows how to reset the index of the DataFrame and drop the old index completely:
Python
0 vote233 views1 answer
Mp.solutions.findhands not in python 3.8
Python
0 vote222 views1 answer
Get localapplication python
Python
0 vote230 views1 answer
Pandas apply pass in arguments
Python
2 votes236 views1 answer
Groups the DataFrame using the specified columns
Python
0 vote250 views1 answer
Scipy check normal distribution
Python