public AttentionPreProcessResult PreProcess(IWeightMatrix inputs, IComputeGraph g)
{
AttentionPreProcessResult r = new AttentionPreProcessResult();
IWeightMatrix bUas = g.RepeatRows(bUa, inputs.Rows);
r.uhs = g.MulAdd(inputs, Ua, bUas);
r.inputs = g.PermuteBatch(inputs, m_batchSize);
return(r);
}
public IWeightMatrix Perform(IWeightMatrix state, AttentionPreProcessResult attenPreProcessResult, IComputeGraph g)
{
var bWas = g.RepeatRows(bWa, state.Rows);
var wc = g.MulAdd(state, Wa, bWas);
var wcs = g.RepeatRows(wc, attenPreProcessResult.inputs.Rows / m_batchSize);
var ggs = g.AddTanh(attenPreProcessResult.uhs, wcs);
var atten = g.Mul(ggs, V);
var atten2 = g.PermuteBatch(atten, m_batchSize);
var attenT = g.Transpose2(atten2);
var attenT2 = g.View(attenT, m_batchSize, attenPreProcessResult.inputs.Rows / m_batchSize);
var attenSoftmax = g.Softmax(attenT2);
IWeightMatrix contexts = g.MulBatch(attenSoftmax, attenPreProcessResult.inputs, m_batchSize);
return(contexts);
}