ray.rllib.utils.tensor_dtype.get_np_dtype

ray.rllib.utils.tensor_dtype.get_np_dtype(x: numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor) → numpy.dtype

Returns the NumPy dtype of the given tensor or array.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.