Fix code snippet in docs where indices should be int32

[nbardy]

The example was returning an error due to invalid tensor type.

---

This change is 

[dsmilkov]

Thanks!

---

Reviewed 1 of 1 files at r1.
Review status: all files reviewed at latest revision, all discussions resolved.

---

Comments from Reviewable

[dsmilkov]

One comment I just noticed tf.oneHot needs the same fix.

---

Review status: all files reviewed at latest revision, all discussions resolved, all commit checks successful.

---

src/ops/array_ops.ts, line 540 at r1 (raw file):

   *
   * ```js
   * tf.oneHot(tf.tensor1d([0, 1]), 3).print();

can you change it here too. oneHot requires indices to be int32?

---

Comments from Reviewable

[dsmilkov]

Hi @nbardy , can you change tf.oneHot in the documentation too?

[manrajgrover]

@dsmilkov I can raise a separate PR for it if it's okay

[dsmilkov]

Thanks! Will merge this in!

---

Review status: all files reviewed at latest revision, 1 unresolved discussion, some commit checks failed.

---

Comments from Reviewable

[dsmilkov]

---

Review status: all files reviewed at latest revision, 1 unresolved discussion, some commit checks failed.

---

Comments from Reviewable