Fixe bug in oneHot doc example (tensorflow#1015)

DOC
HarshaLaxman · May 3, 2018 · 9260c6f · 9260c6f
1 parent bd82d33
commit 9260c6f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/array_ops.ts b/src/ops/array_ops.ts
@@ -546,7 +546,7 @@ export class ArrayOps {
    * `offValue` (defaults to 0).
    *
    * ```js
-   * tf.oneHot(tf.tensor1d([0, 1]), 3).print();
+   * tf.oneHot(tf.tensor1d([0, 1], 'int32'), 3).print();
    * ```
    *
    * @param indices `Tensor1D` of indices with dtype `int32`.