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 vote180 views1 answer
Test if instance in queryset django
Python
6 votes164 views1 answer
Python round number numpy
Python
48 votes167 views6 answers
Pandas replace nan
Python
0 vote183 views4 answers
Python convert dict_keys to list
Python
0 vote181 views1 answer
Python jinja2 from string
Python
81 votes173 views6 answers
Python string to datetime
Python
0 vote167 views1 answer
Class inside class python
Python
5 votes169 views5 answers
Formatted string in python
Python
0 vote175 views1 answer
Pandas join tables based on column of different length
Python
0 vote163 views1 answer
Convert pandas group to dict
Python