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

3 votes201 views3 answers
ModuleNotFoundError: No module named 'MySQLdb' in windows
Python
-1 vote196 views1 answer
Python print empty line
Python
7 votes193 views1 answer
How to truncate values in python to 2 decimal places
Python
0 vote182 views1 answer
Python operator overloading deal with type
Python
9 votes234 views1 answer
Create models in django
Python Django
0 vote147 views1 answer
Pickle.load from gpu device to cpu
Python
0 vote204 views1 answer
Jupyterthemes jplot
Python
0 vote189 views1 answer
Gcsfs pd.read_parquet
Python
13 votes209 views5 answers
Prime number program in python using function
Python
-1 vote271 views1 answer
Ovh minecraft
Python