tf.keras.backend.one_hot

Computes the one-hot representation of an integer tensor.

Compat aliases for migration

tf.compat.v1.keras.backend.one_hot, `tf.compat.v2.keras.backend.one_hot`

tf.keras.backend.one_hot(
    indices, num_classes
)

Arguments
`indices`	nD integer tensor of shape `(batch_size, dim1, dim2, ... dim(n-1))`
`num_classes`	Integer, number of classes to consider.

Returns
(n + 1)D one hot representation of the input with shape `(batch_size, dim1, dim2, ... dim(n-1), num_classes)`

Returns
The one-hot tensor.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/backend/one_hot