I would like to use a weighted MSELoss function for image-to-image training.  I want to specify a weight for each pixel in the target.  Is there a quick/hacky way to do this, or do I need to write my own MSE loss function from scratch?