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 vote238 views1 answer
Get index of max value python numpy
Python
-1 vote284 views1 answer
Random forest classifier in python
Python
0 vote258 views1 answer
Postman authorization
Python
0 vote287 views1 answer
Django.db.utils.ProgrammingError: can't adapt type 'dict_keys'
Python Django
0 vote231 views1 answer
Flask put request
Python
2 votes250 views1 answer
Raise RuntimeError("populate() isn't reentrant")
Python
1 vote261 views1 answer
Polynomial features random forest classifier
Python
0 vote296 views1 answer
Keyword python
Python
54 votes395 views2 answers
How to run django requirement.txt
Python Django
1 vote242 views3 answers
Python sort list by custom function
Python