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 vote294 views4 answers
Download python turtle
Python
0 vote301 views1 answer
Python detect warning
Python
9 votes258 views5 answers
How to drop the index column in pandas
Python
0 vote269 views1 answer
How to check the version of ployly
Python
111 votes259 views6 answers
Rename columns
Python
0 vote291 views1 answer
Django date grater
Python Django
0 vote278 views1 answer
Applies the f function to all Row of this DataFrame
Python
9 votes348 views2 answers
Flask how to run app
Python Flask
0 vote276 views1 answer
How to create an integer validate python
Python
0 vote285 views1 answer
Bouon arrondi tkinter
Python