mindspore.nn.OneHot

class mindspore.nn.OneHot(axis=- 1, depth=1, on_value=1.0, off_value=0.0, dtype=mstype.float32)[源代码]

对输入进行one-hot编码并返回。

输入的 indices 表示的位置取值为on_value,其他所有位置取值为off_value。

Note

如果indices是n阶Tensor,那么返回的one-hot Tensor则为n+1阶Tensor。

如果 indices 是Scalar,则输出shape将是长度为 depth 的向量。

如果 indices 是长度为 features 的向量,则输出shape为:

features * depth if axis == -1

depth * features if axis == 0

如果 indices 是shape为 [batch, features] 的矩阵,则输出shape为:

batch * features * depth if axis == -1

batch * depth * features if axis == 1

depth * batch * features if axis == 0

参数:

  • axis (int) - 指定第几阶为 depth 维one-hot向量,如果轴为-1,则 features * depth ,如果轴为0,则 depth * features 。默认值:-1。

  • depth (int) - 定义one-hot向量的深度。默认值:1。

  • on_value (float) - one-hot值,当 indices[j] = i 时,填充output[i][j]的取值。默认值:1.0。

  • off_value (float) - 非one-hot值,当 indices[j] != i 时,填充output[i][j]的取值。默认值:0.0。

  • dtype (mindspore.dtype) - 是’on_value’和’off_value’的数据类型,而不是输入的数据类型。默认值:mindspore.float32。

输入:

  • indices (Tensor) - 输入索引,任意维度的Tensor,数据类型为int32或int64。

输出:

Tensor,输出Tensor,数据类型 dtype 的one-hot Tensor,维度为 axis 扩展到 depth,并填充on_value和off_value。Outputs 的维度等于 indices 的维度加1。

异常:

  • TypeError - axisdepth 不是int。

  • TypeError - indices 的dtype既不是int32,也不是int64。

  • ValueError - 如果 axis 不在范围[-1, len(indices_shape)]内。

  • ValueError - depth 小于0。

支持平台:

Ascend GPU CPU

样例:

>>> # 1st sample: add new coordinates at axis 1
>>> net = nn.OneHot(depth=4, axis=1)
>>> indices = Tensor([[1, 3], [0, 2]], dtype=mindspore.int32)
>>> output = net(indices)
>>> print(output)
[[[0. 0.]
  [1. 0.]
  [0. 0.]
  [0. 1.]]
 [[1. 0.]
  [0. 0.]
  [0. 1.]
  [0. 0.]]]
>>> # The results are shown below:
>>> print(output.shape)
(2, 4, 2)
>>> # 2nd sample: add new coordinates at axis 0
>>> net = nn.OneHot(depth=4, axis=0)
>>> indices = Tensor([[1, 3], [0, 2]], dtype=mindspore.int32)
>>> output = net(indices)
>>> print(output)
[[[0. 0.]
  [1. 0.]]
 [[1. 0.]
  [0. 0.]]
 [[0. 0.]
  [0. 1.]]
 [[0. 1.]
  [0. 0.]]]
>>> # The results are shown below:
>>> print(output.shape)
(4, 2, 2)
>>> # 3rd sample: add new coordinates at the last dimension.
>>> net = nn.OneHot(depth=4, axis=-1)
>>> indices = Tensor([[1, 3], [0, 2]], dtype=mindspore.int32)
>>> output = net(indices)
>>> # The results are shown below:
>>> print(output)
[[[0. 1. 0. 0.]
  [0. 0. 0. 1.]]
 [[1. 0. 0. 0.]
  [0. 0. 1. 0.]]]
>>> print(output.shape)
(2, 2, 4)
>>> indices = Tensor([1, 3, 0, 2], dtype=mindspore.int32)
>>> output = net(indices)
>>> print(output)
[[0. 1. 0. 0.]
 [0. 0. 0. 1.]
 [1. 0. 0. 0.]
 [0. 0. 1. 0.]]
>>> print(output.shape)
(4, 4)