Python mxnet.ndarray.one_hot用法及代码示例

用法:
mxnet.ndarray.one_hot(indices=None, depth=_Null, on_value=_Null, off_value=_Null, dtype=_Null, out=None, name=None, **kwargs)

参数：

indices：(NDArray) - 设置on_value的位置数组
depth：(int, required) - 一个热维度的深度。
on_value：(double, optional, default=1) - 分配给由索引表示的位置的值。
off_value：(double, optional, default=0) - 分配给索引未表示的位置的值。
dtype：({'bfloat16', 'float16', 'float32', 'float64', 'int32', 'int64', 'int8', 'uint8'},optional, default='float32') - 输出的 DType
out：(NDArray, optional) - 输出 NDArray 来保存结果。

out：- 此函数的输出。

返回类型：

NDArray 或 NDArray 列表

返回 one-hot 数组。

indices 表示的位置取值 on_value ，而所有其他位置取值 off_value 。

one_hot 操作与形状 (i0, i1) 的 indices 和 d 的 depth 将导致形状为 (i0, i1, d) 的输出数组：

output[i,j,:] = off_value
output[i,j,indices[i,j]] = on_value

例子：

one_hot([1,0,2,0], 3) = [[ 0.  1.  0.]
                         [ 1.  0.  0.]
                         [ 0.  0.  1.]
                         [ 1.  0.  0.]]

one_hot([1,0,2,0], 3, on_value=8, off_value=1,
        dtype='int32') = [[1 8 1]
                          [8 1 1]
                          [1 1 8]
                          [8 1 1]]

one_hot([[1,0],[1,0],[2,0]], 3) = [[[ 0.  1.  0.]
                                    [ 1.  0.  0.]]

                                   [[ 0.  1.  0.]
                                    [ 1.  0.  0.]]

                                   [[ 0.  0.  1.]
                                    [ 1.  0.  0.]]]

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.one_hot。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：