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 vote319 views1 answer
Arabic text recognition from pdf using python
Python
-1 vote243 views2 answers
Python dict remove duplicates where name are not the same
Python
5 votes205 views5 answers
Convert dictionary keys to list python
Python
0 vote227 views1 answer
How to let something see on screen with python
Python
8 votes201 views2 answers
Python requests header
Python
4 votes218 views2 answers
Split filename and extension python
Python
0 vote346 views1 answer
Convert method to str python
Python
13 votes204 views2 answers
How to copy one dictionary to another in python
Python
12 votes195 views3 answers
Convert pandas dataframe to dict with a column as key
Python
3 votes220 views4 answers
Convert from python to javascript online
Python