torch:cat

(torch:cat tensors &key axis)

Differentiable concatenation of the list tensors along an existing axis (linalg:concatenate, torch.cat; default 0, negative counts from the end). The backward pass slices the gradient back into each input's extent along that axis.