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 vote190 views1 answer
Split string and convert to int python
Python
0 vote174 views2 answers
How to find common characters in two strings in python
Python
3 votes199 views1 answer
No module named 'django.core.urlresolvers'
Python
2 votes231 views1 answer
Pandas series plot horizontal bar
Python
1 vote196 views1 answer
Discord.py commands.group
Python
0 vote192 views1 answer
Django model specify table name
Python Django
20 votes182 views4 answers
Python remove new line
Python
0 vote159 views1 answer
UTC to ISO 8601 with Local TimeZone information without microsecond (Python 3):
Python
0 vote173 views1 answer
Python evaluate string
Python
0 vote176 views1 answer
Selenium python has no attrirute getText
Python