ray.rllib.core.learner.learner.Learner._get_tensor_variable
ray.rllib.core.learner.learner.Learner._get_tensor_variable#
- abstract Learner._get_tensor_variable(value: Any, dtype: Any = None, trainable: bool = False) Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor][source]#
Returns a framework-specific tensor variable with the initial given value.
This is a framework specific method that should be implemented by the framework specific sub-class.
- Parameters
value – The initial value for the tensor variable variable.
- Returns
The framework specific tensor variable of the given initial value, dtype and trainable/requires_grad property.