详解mxnet.ndarray.one_hot。即转变为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) : 要转化为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.]]]

  参考:这里

你可能感兴趣的:(mxnet)