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

13 votes270 views1 answer
Configure funCtion in tkinter
Python
0 vote264 views2 answers
Django create view filter options
Python
19 votes277 views1 answer
Python string cut left
Python
3 votes264 views1 answer
How to change python version on linux
Python
0 vote267 views1 answer
Compile python folder
Python
0 vote258 views1 answer
Pytest installation windows
Python
21 votes272 views1 answer
How to plot a graph using matplotlib
Python
0 vote278 views1 answer
Python best docstring format
Python
1 vote297 views2 answers
AttributeError: module 'yaml' has no attribute 'load_all'
Python
0 vote261 views1 answer
For in list start with index python
Python