Fix the lstm pseudo sample code

webmachinelearning · May 8, 2023 · d4d8148 · d4d8148
1 parent c8f65e3
commit d4d8148
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/index.bs b/index.bs
@@ -2154,8 +2154,8 @@ partial interface MLGraphBuilder {
         let output = builder.reshape(results[0], [1, null, hiddenSize]);
         let cell = builder.reshape(results[1], [1, null, hiddenSize]);
 
-        nextHidden = (nextHidden ? builder.concat([nextHidden, result], 0) : output);
-        nextCell = (nextCell ? builder.concat([nextCell, result], 0) : cell);
+        nextHidden = (nextHidden ? builder.concat([nextHidden, output], 0) : output);
+        nextCell = (nextCell ? builder.concat([nextCell, cell], 0) : cell);
       }
 
       hiddenState = nextHidden;