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 vote335 views1 answer
H2o dataframe columns drop
Python
27 votes318 views3 answers
How to convert exe file to python file
Python
0 vote302 views1 answer
Python list files in folder with wildcard
Python
0 vote326 views1 answer
NameError: name 'CAP_PROP_FPS' is not defined
Python
0 vote315 views4 answers
Python draw circle matplotlib
Python
1 vote298 views1 answer
How to refresh windows 10 with python
Python
0 vote354 views1 answer
Instabot login python
Python
0 vote305 views1 answer
How to deal with SettingWithCopyWarning in Pandas
Python
1 vote312 views2 answers
Python turn off garbage collection
Python
0 vote317 views1 answer
Sum of positive numbers in array with negative python
Python