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 vote329 views1 answer
How to run django in jupyter
Python Django
-1 vote278 views1 answer
Date and time using tkinter
Python
2 votes315 views1 answer
Pandas dataframe set datetime index
Python
2 votes360 views1 answer
Django try catch exception
Python Django
1 vote321 views1 answer
Cookies in django
Python Django
0 vote293 views1 answer
Python count elements in sublists
Python
3 votes344 views2 answers
Python subprocess exception handling
Python
1 vote268 views1 answer
Add text toimage cv2
Python
0 vote270 views1 answer
Find highest value in array python
Python
1 vote280 views1 answer
Error: can't copy 'brz.1': doesn't exist or not a regular file
Python