ray.rllib.policy.eager_tf_policy_v2.EagerTFPolicyV2.maybe_remove_time_dimension

EagerTFPolicyV2.maybe_remove_time_dimension(input_dict: Dict[str, numpy.array | jnp.ndarray | tf.Tensor | torch.Tensor])

Removes a time dimension for recurrent RLModules.

Parameters:

input_dict – The input dict.

Returns:

The input dict with a possibly removed time dimension.