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 vote261 views1 answer
Mudopy
Python
4 votes272 views1 answer
Get prime number python
Python
0 vote267 views2 answers
How to make a time limit using renpy
Python
0 vote263 views1 answer
Python permission denied on mac
Python
0 vote253 views1 answer
Time.ctime(os.path.getmtime phyton in datetime
Python
0 vote274 views1 answer
Python get portion of dictionary
Python
0 vote276 views10 answers
EMAIL_BACKEND = 'django.core.mail.backends.console.EmailBackend'
Python Django
0 vote286 views1 answer
RuntimeError: Broken toolchain: cannot link a simple C program
Python
-1 vote265 views1 answer
Python object is not subscriptable
Python
0 vote267 views1 answer
How to get a summary of a column in python
Python