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 vote107 views1 answer
What modules are used for NLG in python
Python
0 vote100 views1 answer
Stack widgets in tkinter
Python
0 vote105 views1 answer
Append path to sys jupyter notebook
Python
0 vote103 views1 answer
Python scheduler
Python
0 vote102 views1 answer
Python selenium disable JavaScript Detection
Python
14 votes112 views4 answers
Pip install virtualenv windows
Python
-1 vote82 views1 answer
Image analysis python
Python
2 votes103 views1 answer
Confidence intervals in python
Python
1 vote98 views1 answer
Pyhton built in sort
Python
2 votes120 views4 answers
How to do fibonacci sequence in python
Python