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

1 vote202 views1 answer
Python get last item in a list
Python
0 vote175 views1 answer
Load image metadata with pil
Python
0 vote212 views1 answer
Python beautifulsoup load cookies download file from url
Python
0 vote186 views1 answer
Minimumm sort tryed python
Python
0 vote182 views2 answers
How to plot two different class in different colour in python
Python
52 votes173 views6 answers
Python new line
Python
1 vote188 views1 answer
How to convert 24 hours to 12 hours in python
Python
1 vote175 views1 answer
Save image requests python
Python
0 vote190 views1 answer
How to draw a single pixel in pyglet
Python
1 vote232 views1 answer
How do i check if a django queryset is empty
Python Django