ray.rllib.utils.torch_utils.convert_to_torch_tensor#
- ray.rllib.utils.torch_utils.convert_to_torch_tensor(x: numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor | dict | tuple, device: str | None = None)[source]#
Converts any struct to torch.Tensors.
- x: Any (possibly nested) struct, the values in which will be
converted and returned as a new struct with all leaves converted to torch tensors.
- Returns:
A new struct with the same structure as
x
, but with all values converted to torch Tensor types. This does not convert possibly nested elements that are None because torch has no representation for that.- Return type:
Any