Hello,
Let’s assume I have a Tensor of shape [M , D] where M is the number of samples and D is the feature dimension. additionally a mask tensor of shape [M , K] which contains K different indices for each of the M samples.
my question is how can I for each of the M samples calculate the mean of its K indices (for each sample we have in the mask which K indices we need to use)
Thanks!