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 vote372 views3 answers
Python code to save data with multiple sheet in excel
Python
0 vote362 views1 answer
Can we use python functions in node
Python
80 votes333 views3 answers
Python remove last part of string
Python
0 vote340 views1 answer
Python if string is null or whitespace
Python
1 vote333 views1 answer
Python condition if dataype
Python
-2 votes344 views1 answer
Play sound directly with pytts
Python
6 votes353 views6 answers
Python tuples
Python
0 vote370 views1 answer
The requested resource was not found on this server. django
Python Django
0 vote353 views1 answer
Flask - store object directly in a session [duplicate]
Python
0 vote331 views1 answer
Flask mongoengine configuration
Python