ray.rllib.models.distributions.Distribution.required_input_dim#

abstract static Distribution.required_input_dim(space: gymnasium.Space, **kwargs) → int[source]#

Returns the required length of an input parameter tensor.

Parameters:

  • space – The space this distribution will be used for, whose shape attributes will be used to determine the required shape of the input parameter tensor.

  • **kwargs – Forward compatibility placeholder.

Returns:

size of the required input vector (minus leading batch dimension).