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

28 votes358 views2 answers
Get a slice of string in python
Python
0 vote315 views1 answer
Pandas apply pass in arguments
Python
0 vote328 views1 answer
Smtp django
Python
0 vote399 views2 answers
Find All Occurrences of a Substring in a String in Python
Python
0 vote299 views1 answer
Python 'encoding' is an invalid keyword argument for this function"
Python
0 vote320 views2 answers
Python normalized correlation
Python
0 vote328 views1 answer
Pandas normalize rows to max value
Python
14 votes316 views3 answers
Python remove first and last character from string
Python
0 vote317 views1 answer
Remove stopwords from list of strings python
Python
3 votes353 views2 answers
Multiple Regression in Python
Python