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

5 votes314 views6 answers
Valid Palindrome
Python
4 votes308 views1 answer
Bold text variable in python
Python
213 votes584 views6 answers
Python iterate through dictionary
Python
41 votes299 views4 answers
Find out current datetime in python
Python
0 vote276 views1 answer
Python - calculate the value range on a df
Python
0 vote293 views1 answer
Performance of extend vs append loop
Python
0 vote304 views1 answer
Remote python running line by line visual code
Python
0 vote287 views1 answer
Exec inside def is not working in python
Python
2 votes332 views3 answers
Matplotlib show plot
Python
31 votes294 views2 answers
How to divide string in python
Python