ray.rllib.core.learner.learner.Learner._get_tensor_variable

abstract Learner._get_tensor_variable(value: Any, dtype: Any = None, trainable: bool = False) → numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor

返回一个特定于框架的张量变量，并赋予初始给定的值。

这是一个特定于框架的方法，应由特定于框架的子类实现。

参数:

value – 张量变量 variable 的初始值。

返回:

特定框架的张量变量，具有给定的初始值、数据类型和可训练/需要梯度属性。