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

2 votes172 views1 answer
Python dict exclude keys
Python
0 vote152 views1 answer
How to let the user input desmials in python
Python
0 vote175 views1 answer
Bs.newtag() inner html
Python
1 vote196 views4 answers
Pandas count zeros in column
Python
2 votes144 views1 answer
Get email address from string python
Python
0 vote204 views1 answer
Loading model
Python
2 votes169 views1 answer
Np logical_or
Python
4 votes169 views2 answers
How to plot two columns graphs in python
Python
0 vote164 views2 answers
Error: Could not locate a Flask application. You did not provide the "FLASK_APP" environment variable, and a "wsgi.py" or "app.py" module was not found in the current directory.
Python
0 vote170 views1 answer
Pyevtk documentation writearraystovtk
Python