FlattenConcatStateValueNet¶
-
class
maze.perception.models.built_in.flatten_concat.
FlattenConcatStateValueNet
(*args: Any, **kwargs: Any)¶ Flatten and concatenation state value model.
- Parameters
obs_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.