tensorflow API: tf.one_hot

最新推荐文章于 2022-06-02 15:27:35 发布

NockinOnHeavensDoor

最新推荐文章于 2022-06-02 15:27:35 发布

阅读量363

点赞数

分类专栏： tensorflow 文章标签： api one_hot

tensorflow 专栏收录该内容

64 篇文章 0 订阅

订阅专栏

本文详细介绍了 TensorFlow 中 one_hot 函数的用法及其参数设置，包括 indices、depth、on_value 和 off_value 的含义，并通过多个示例展示了如何使用该函数进行独热编码。

摘要由CSDN通过智能技术生成

one_hot(indices, depth, on_value=None, off_value=None, axis=None, dtype=None, name=None)
参数:
indices: 需要进行独特编码的张量。
depth: 独热编码的长度。
on_value: 在独热处默认是编码为1(default: 1)
off_value: 默认使用0作为填充值(default: 0)
axis: 默认-1，表示最内部的轴进行。
dtype: 输出tensor的数据类型

例子：

  ```python
    indices = [0, 1, 2]
    depth = 3
    tf.one_hot(indices, depth)  # output: [3 x 3]
    # [[1., 0., 0.],
    #  [0., 1., 0.],
    #  [0., 0., 1.]]

    indices = [0, 2, -1, 1]
    depth = 3
    tf.one_hot(indices, depth,
               on_value=5.0, off_value=0.0,
               axis=-1)  # output: [4 x 3]
    # [[5.0, 0.0, 0.0],  # one_hot(0)
    #  [0.0, 0.0, 5.0],  # one_hot(2)
    #  [0.0, 0.0, 0.0],  # one_hot(-1)
    #  [0.0, 5.0, 0.0]]  # one_hot(1)

  indices = [[0, 2], [1, -1]]
    depth = 3
    tf.one_hot(indices, depth,
               on_value=1.0, off_value=0.0,
               axis=-1)  # output: [2 x 2 x 3]
    # [[[1.0, 0.0, 0.0],   # one_hot(0)
    #   [0.0, 0.0, 1.0]],  # one_hot(2)
    #  [[0.0, 1.0, 0.0],   # one_hot(1)
    #   [0.0, 0.0, 0.0]]]  # one_hot(-1)
    ```