Fix code snippet in docs where indices should be int32 (tensorflow#989)

DOC
indutny · May 3, 2018 · 52cc2f6 · 52cc2f6
1 parent c6a6950
commit 52cc2f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ops/array_ops.ts b/src/ops/array_ops.ts
@@ -867,14 +867,14 @@ export class ArrayOps {
    *
    * ```js
    * const x = tf.tensor1d([1, 2, 3, 4]);
-   * const indices = tf.tensor1d([1, 3, 3]);
+   * const indices = tf.tensor1d([1, 3, 3], 'int32');
    *
    * x.gather(indices).print();
    * ```
    *
    * ```js
    * const x = tf.tensor2d([1, 2, 3, 4], [2, 2]);
-   * const indices = tf.tensor1d([1, 1, 0]);
+   * const indices = tf.tensor1d([1, 1, 0], 'int32');
    *
    * x.gather(indices).print();
    * ```