它有这些参数。
mxnet.ndarray.one_hot(indices=None, depth=_Null, on_value=_Null, off_value=_Null, dtype=_Null, out=None, name=None, **kwargs)
解释一下重要的参数:
indices (NDArray) : 要转化为one—hot的数据。
depth (int, required) : one-hot的维度
on_value (double, optional, default=1) : 分配给由索引表示的位置的值。
off_value (double, optional, default=0) :不是索引位置的值。
例子:
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) = [[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.]]]
参考:这里