oa::FnMatrix::gatherBwd

Function published by the oa::FnMatrix namespace.

Function Documentation

Matrix oa::FnMatrix::gatherBwd( const Matrix & inIndices, const Matrix & inGradOutput, oa::I32 inVocabSize, oa::I32 inEmbedDim )
No public source comment is attached to this declaration.

Parameters

inIndices
const Matrix &

inGradOutput
const Matrix &

inVocabSize
oa::I32

inEmbedDim
oa::I32

Returns

Matrix

The declared return value.