Motivation

While there is the select_out_keys method #350 I would like to select out keys directly while initialising a sequential module.
The ProbabilisticTensorDictSequential does not support the selected_out_keys argument of it's parent TensorDictSequential.

Solution

Add the selected_out_keys argument to the __init__ of ProbabilisticTensorDictSequential and pass it through to the __init__ of the parent.

Alternatives

The current workaround is to modify the instance of ProbabilisticTensorDictSequential after initialisation with select_out_keys.

Additional context

Checklist

• I have checked that there is no similar issue in the repo (required)