fn
mse_loss_op
→TensorImplPtrint mse_loss_op(const int & input, const int & target, int reduction)Python wrappers
Public Python APIs implemented by this engine symbol.Public mean-squared-error entry point.
Thin wrapper that casts reduction to Reduction and delegates to
MseLossBackward::forward.
Parameters
inputTensorImplPtrPredicted tensor.
targetTensorImplPtrGround-truth tensor of the same shape and dtype.
reductionint0 (None), 1 (Mean), or 2 (Sum).Returns
TensorImplPtrScalar (Mean / Sum) or per-element (None) loss.