FlattenConcatPolicyNet

class maze.perception.models.built_in.flatten_concat.FlattenConcatPolicyNet(*args: Any, **kwargs: Any)

Flatten and concatenation policy model.

Parameters:

• obs_shapes – Dictionary mapping of observation names to shapes.

• action_logits_shapes – Dictionary mapping of observation names to shapes.

• hidden_units – Dictionary mapping of action names to shapes.

• non_lin – The non-linearity to apply.

forward(x)

forward pass.