I would like to provide custom weights for each sample that I evaluate the loss for in a batch.
At the moment I can do:
F.cross_entropy(y_pred, y_true, weight=weight)

and this will use a single set of weights for the entire batch.
What I would like to do is pass in a set of weights for each sample in the batch