oa::FnLoss::maskedCrossEntropyBwd

maskedCrossEntropyBwd: backward for maskedCrossEntropy.

Function Documentation

Matrix oa::FnLoss::maskedCrossEntropyBwd( const Matrix & inLogits, const Matrix & inTargets, const Matrix & inMask, oa::I32 inValidCount )
maskedCrossEntropyBwd: backward for maskedCrossEntropy.

Parameters

inLogits
const Matrix &

inTargets
const Matrix &

inMask
const Matrix &

inValidCount
oa::I32

Returns

Matrix

The declared return value.