| View source on GitHub |
Computes the one-hot representation of an integer tensor.
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/r2.3/api_docs/python/tf/keras/backend/one_hot