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

1 vote147 views1 answer
How to get index of week in list in python
Python
0 vote254 views1 answer
Bulk upload with dictionary or list in django moels
Python Django
0 vote145 views1 answer
Size of the query process in python BigQuery
Python
12 votes148 views4 answers
Get index of all element in list python
Python
0 vote157 views1 answer
Python how to convert a list of floats to a list of strings
Python
2 votes155 views2 answers
Tkinter mainloop
Python
0 vote142 views1 answer
Pytube sample script
Python
1 vote140 views2 answers
Get definition of word python
Python
0 vote145 views1 answer
Set .intersection() operation solution in python3
Python
2 votes147 views2 answers
Requests sessions
Python