Hi!
I am working with a model in which I would like to perform the following operation:
Z is an n x m matrix
S is an k x l matrix
I am searching for a pytorch operation that would allow me to concatenate each pair of rows in Z and columns in S to yield a 3D tensor, where entry i,j contains row i in z concatenated with column j in S and hence has length (m + k). Hence, the tensor dimensions would be n x l x (m + k). Is there a convenient way to do this in pytorch?
Thx!