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

2 votes132 views2 answers
Ticks font size matplotlib
Python
0 vote114 views1 answer
Cv2 opencv-python imshow while loop
Python
0 vote113 views1 answer
Calculate term frequency python
Python
9 votes120 views6 answers
Python text to speech
Python
-2 votes121 views1 answer
En_core_web_sm
Python
0 vote115 views1 answer
Python absolute path from projectr
Python
0 vote121 views1 answer
Python parsing a .srt file with regex - Remove all, keep only text
Python
0 vote112 views1 answer
Optimize images using pillow
Python
0 vote114 views1 answer
Python How To Convert Text to Speech
Python
84 votes125 views6 answers
Sort dataframe by column
Python