Description

Enable dispatching arguments for the .forward() method of the PPO loss module.

Types of changes

What types of changes does your code introduce? Remove all that do not apply:

• Breaking change (fix or feature that would cause existing functionality to change)
• Documentation (update in the documentation)

Checklist

Go over all the following points, and put an x in all the boxes that apply.
If you are unsure about any of these, don't hesitate to ask. We are here to help!

• I have read the CONTRIBUTION guide (required)
• I have updated the tests accordingly (required for a bug fix or a new feature).
• I have updated the documentation accordingly.