Hi.
I have the following setup:
[49, 49] matrix, where each row is a probabilities vector (obtained from softmax over logits). overall it has 49 probability vectors, each with 49 examples.
[49, x, y] matrix, containig 49 spectrograms of size [x,y] each.
I try to obtain a 49 different weighted spectrograms, from each of the 49 probability vectors and 49 spectrograms.
Output size shall be [49, x, y].
I tried my best to search the net, and tried many configurtations of torch matmul, bmm, etc…