# Module: tf_agents.trajectories.policy_step

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/agents/blob/v0.19.0/tf_agents/trajectories/policy_step.py) |

Policy Step.

## Classes

[`class CommonFields`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/policy_step/CommonFields): Strings which can be used for querying returned PolicyStep.info field.

[`class PolicyInfo`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/PolicyInfo): PolicyInfo(log_probability,)

[`class PolicyStep`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/PolicyStep): Returned with every call to `policy.action()` and `policy.distribution()`.

## Functions

[`get_log_probability(...)`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/policy_step/get_log_probability): Gets the CommonFields.LOG_PROBABILITY from info depending on type.

[`set_log_probability(...)`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/policy_step/set_log_probability): Sets the CommonFields.LOG_PROBABILITY on info to be log_probability.

## Type Aliases

[`ActionType`](https://www.tensorflow.org/agents/api_docs/python/tf_agents/trajectories/policy_step/ActionType)

<br />

<br />

<br />

| ## Other Members ||
|---|---|
| absolute_import | Instance of `__future__._Feature` |
| division | Instance of `__future__._Feature` |
| print_function | Instance of `__future__._Feature` |

<br />