Stack a and b along a new dimension inserted at position dim. This adds an extra dimension to the output, rather than joining the tensors along an existing one.
a
b
dim
Sign in to write and run your own code.
solve(...)