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…