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.