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:nD整数张量,具有shape (batch_size, dim1, dim2, ... dim(n-1))
  • num_classes:整数,要考虑的类别数。

返回:

输入的(n + 1)D one-hot表示,shape为 (batch_size, dim1, dim2, ... dim(n-1), num_classes)。

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部