ray.rllib.utils.numpy.one_hot

ray.rllib.utils.numpy.one_hot(x: numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor | int, depth: int = 0, on_value: float = 1.0, off_value: float = 0.0, dtype: type = numpy.float32) → numpy.ndarray

用于numpy的独热编码实用函数。

感谢 qianyizhang: https://gist.github.com/qianyizhang/07ee1c15cad08afb03f5de69349efc30.

参数:

• x – 要进行独热编码的输入。

• depth – 要进行独热编码的最大数量（最后一级的尺寸）。

• on_value – 用于开启的值。默认值：1.0。

• off_value – 用于关闭的值。默认值：0.0。

返回:

输入数组的热编码等价形式。