Fix incorrect step reference.

This commit is contained in:
Ryan Curtin
2022-03-27 14:16:58 -04:00
parent 7fa7eb15f0
commit f2a1b79e99
+1 -1
View File
@@ -267,7 +267,7 @@ void LSTMType<InputType, OutputType>::Forward(
if (this->HasPreviousStep())
{
outputGate +=
output2GateOutputWeight * outParameter.slice(this->CurrentStep());
output2GateOutputWeight * outParameter.slice(this->PreviousStep());
}
outputGate.each_col() += input2GateOutputBias;