Masking regions of output

hi pytorch
I have a fully convolutional network that operates on some training data that contains invalid values. How can I mask out these regions for the purpose of calculating loss, given the binary masks?

you can mask by:

mask= (output!=invalid_value)
mask_target=target[mask]
mask_output=output[mask]
loss = calculate_loss(mask_output,mask_target)