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 vote143 views1 answer
How to compress image field in django?
Python Django
9 votes111 views1 answer
Print current dirfile dir python
Python
1 vote114 views1 answer
Subprocess the system cannot find the file specified
Python
1 vote110 views3 answers
For j in range python
Python
1 vote119 views1 answer
Keras.layers.MaxPool2D
Python
14 votes115 views5 answers
Server in python
Python
25 votes115 views2 answers
Start new app in django
Python
2 votes117 views2 answers
What is role of ALLOWED_HOSTs in Django
Python
0 vote123 views1 answer
IsADirectoryError: [Errno 21] Is a directory: '/'
Python
3 votes107 views3 answers
Pandas merge two dataframes remove duplicates
Python