Fixed EuclideanMaskedLossLayer + GPU implementation