For a tensor a in shape of 1 x n, and b in batch x m x n, is there any elegant way to concatenate a to the second dimension of b so that the result should in shape of batch x (m + 1) x n. Meanwhile, the gradient in different batch can effect the same origin a by loss.backward().