Python mxnet.ndarray.op.one_hot用法及代碼示例

用法:
mxnet.ndarray.op.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.op.one_hot。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：