# 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)

Returns a one-hot array.

The locations represented by indices take value on_value, while all other locations take value off_value.

one_hot operation with indices of shape (i0, i1) and depth of d would result in an output array of shape (i0, i1, d) with:

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


Examples:

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.]]]


Defined in src/operator/tensor/indexing_op.cc:L796

Parameters: indices (NDArray) – array of locations where to set on_value depth (int, required) – Depth of the one hot dimension. on_value (double, optional, default=1) – The value assigned to the locations represented by indices. off_value (double, optional, default=0) – The value assigned to the locations not represented by indices. dtype ({'float16', 'float32', 'float64', 'int32', 'int64', 'int8', 'uint8'},optional, default='float32') – DType of the output out (NDArray, optional) – The output NDArray to hold the result. out – The output of this function. NDArray or list of NDArrays