TensorFlow函數(shù)教程：tf.keras.backend.one_hot

tf.keras.backend.one_hot函數(shù)

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

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

計(jì)算一個整數(shù)張量的 one-hot 表示。

參數(shù)：

• indices：整數(shù)張量，尺寸為 (batch_size, dim1, dim2, ... dim(n-1))
• num_classes：整數(shù)，需要考慮的類別數(shù)。

返回：

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