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 vote230 views1 answer
COLLECTING
Python
21 votes282 views2 answers
Getting pi in python
Python
0 vote231 views1 answer
Instead of: firstName = "John" lastName = "Henry" city = "Manchester"
Python
16 votes208 views1 answer
Super idol
Python
13 votes211 views6 answers
How to define a dictionary in python
Python
2 votes229 views2 answers
Add text to plot python
Python
1 vote216 views2 answers
Unknown amount of arguments discord py
Python
0 vote237 views1 answer
Python Textfeld lköschen
Python
2 votes237 views2 answers
Panda - subset based on column value
Python
0 vote258 views1 answer
Travis deployment script for django applications to heroku
Python Django