TensorFlow函数教程:tf.keras.backend.one_hot
2020-09-04 10:02 更新
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)
。