ray.rllib.models.distributions.Distribution.required_input_dim

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

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).