Fix IndexOutOfBoundsException in the Dot layer (#357)

Kotlin · Apr 19, 2022 · efaa1eb · efaa1eb
1 parent 1f4d678
commit efaa1eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/merge/Dot.kt
@@ -30,7 +30,7 @@ public class Dot(
         var x2 = input[1]
         val axes: IntArray = IntArray(2)
         val scope: Scope = tf.scope()
-        for (i in 0..2) {
+        for (i in 0 until 2) {
             if (axis[i] < 0) {
                 axes[i] = axis[i] % input[i].asOutput().shape().numDimensions()
             } else {