nexuml_library.layers.loss.regression_loss¶
Regression loss layer.
Bases: LayerDefinition
Computes MSE regression loss between predictions and targets.
keys_in: [predictions_key] keys_out: [loss_key] Reads targets from y TensorDict using label_key.