oa::FnLoss::maskedCrossEntropy

maskedCrossEntropy: cross-entropy over only rows where inMask is non-zero.

Function Documentation

Matrix oa::FnLoss::maskedCrossEntropy( const Matrix & inLogits, const Matrix & inTargets, const Matrix & inMask, oa::I32 inValidCount )
maskedCrossEntropy: cross-entropy over only rows where inMask is non-zero.

Parameters

inLogits
const Matrix &

inTargets
const Matrix &

inMask
const Matrix &

inValidCount
oa::I32

Returns

Matrix

The declared return value.