Given data and mask tensors are there a pytorch-way to obtain masked aggregations of data (mean, max, m