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

5 votes312 views4 answers
Install python venv install
Python
1 vote331 views1 answer
What is the meaning of illiteral with base 10
Python
1 vote322 views1 answer
How to get sum specific columns value in machine learning
Python
0 vote312 views1 answer
Im save to a bytes io python
Python
1 vote330 views1 answer
Displays unique data including null data
Python
0 vote326 views1 answer
Pandas index dataframe with integers
Python
2 votes379 views3 answers
Play mp3 file python
Python
6 votes306 views2 answers
Python reverse words and swap case
Python
0 vote324 views1 answer
How to find 2 similar words in atext python
Python
-1 vote301 views2 answers
Sequence python
Python