TensorFlow函数教程：tf.keras.backend.one_hot

tf.keras.backend.one_hot函数

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

定义在：tensorflow/python/keras/backend.py。

计算一个整数张量的 one-hot 表示。

参数：

• indices：整数张量，尺寸为 (batch_size, dim1, dim2, ... dim(n-1))
• num_classes：整数，需要考虑的类别数。

返回：

输入的（n + 1）D one-hot 表示，尺寸为 (batch_size, dim1, dim2, ... dim(n-1), num_classes)。