ray.rllib.utils.numpy.convert_to_numpy

ray.rllib.utils.numpy.convert_to_numpy(x: numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor | dict | tuple, reduce_type: bool = True) → numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor | dict | tuple

将 stats 中的值转换为非张量的 numpy 或 python 类型。

参数:

• x – 任何（可能是嵌套的）结构，其中的值将被转换并以新的结构返回，其中所有 torch/tf 张量将被转换为 numpy 类型。

• reduce_type – 是否自动将所有 float64 和 int64 数据分别减少为 float32 和 int32 数据。

返回:

一个与 x 结构相同的新结构体，但所有值都转换为 numpy 数组（在 CPU 上）。