ray.rllib.policy.eager_tf_policy_v2.EagerTFPolicyV2.action_distribution_fn
ray.rllib.policy.eager_tf_policy_v2.EagerTFPolicyV2.action_distribution_fn#
- EagerTFPolicyV2.action_distribution_fn(model: ray.rllib.models.modelv2.ModelV2, *, obs_batch: Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor], state_batches: Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor], **kwargs) Tuple[Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor], type, List[Union[numpy.array, jnp.ndarray, tf.Tensor, torch.Tensor]]] [source]#
Action distribution function for this Policy.
- Parameters
model – Underlying model.
obs_batch – Observation tensor batch.
state_batches – Action sampling state batch.
- Returns
Distribution input. ActionDistribution class. State outs.