用法:
mxnet.symbol.op.one_hot(indices=None, depth=_Null, on_value=_Null, off_value=_Null, dtype=_Null, name=None, attr=None, out=None, **kwargs)
- indices:(
Symbol
) - 设置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 - name:(
string
,
optional.
) - 结果符号的名称。
- indices:(
结果符号。
参数:
返回:
返回类型:
返回 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.]]]
相关用法
- Python mxnet.symbol.op.ones_like用法及代码示例
- Python mxnet.symbol.op.broadcast_logical_xor用法及代码示例
- Python mxnet.symbol.op.log_softmax用法及代码示例
- Python mxnet.symbol.op.SliceChannel用法及代码示例
- Python mxnet.symbol.op.linalg_trmm用法及代码示例
- Python mxnet.symbol.op.broadcast_plus用法及代码示例
- Python mxnet.symbol.op.broadcast_mul用法及代码示例
- Python mxnet.symbol.op.reciprocal用法及代码示例
- Python mxnet.symbol.op.argmax用法及代码示例
- Python mxnet.symbol.op.linalg_syrk用法及代码示例
- Python mxnet.symbol.op.erf用法及代码示例
- Python mxnet.symbol.op.broadcast_minus用法及代码示例
- Python mxnet.symbol.op.random_pdf_negative_binomial用法及代码示例
- Python mxnet.symbol.op.linalg_gelqf用法及代码示例
- Python mxnet.symbol.op.round用法及代码示例
- Python mxnet.symbol.op.sample_normal用法及代码示例
- Python mxnet.symbol.op.swapaxes用法及代码示例
- Python mxnet.symbol.op.topk用法及代码示例
- Python mxnet.symbol.op.erfinv用法及代码示例
- Python mxnet.symbol.op.floor用法及代码示例
注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.symbol.op.one_hot。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。