ray.rllib.models.tf.tf_modelv2.TFModelV2.value_function

TFModelV2.value_function() → Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor]

Returns the value function output for the most recent forward pass.

Note that a forward call has to be performed first, before this methods can return anything and thus that calling this method does not cause an extra forward pass through the network.

Returns

Value estimate tensor of shape [BATCH].