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

4 votes129 views2 answers
Python import file from parent directory
Python
6 votes157 views2 answers
How to get the current web page link in selenium pthon
Python
12 votes143 views2 answers
How to create dynamic variable names in python
Python
19 votes176 views6 answers
Pop function in python
Python
2 votes149 views2 answers
Python how to sum two lists
Python
0 vote174 views1 answer
Django clear _pycache_ command
Python
3 votes150 views2 answers
Python find largest variable
Python
0 vote152 views1 answer
Django forms request
Python
2 votes173 views1 answer
Django static media
Python
51 votes200 views6 answers
Reverse array
Python