I have two lists of matrices. The first one has shape [(N, n1), (N, n2), (N, n3)]. The second of has shape [(n1, B), (n2, B), (n3, B)]. I would like to multiply the two lists to obtain [(N, B), (N,B), (N, B)]. Is it possbile to do this in Pytorch without looping through the lists?
Thank you very much!