Module Owl_symbolic_neural_graph
val init : Owl_symbolic_types.nn_init -> int array -> Owl_symbolic_graph.symbolval activation : ?name:string -> Owl_symbolic_types.activation -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval input : ?name:string -> ?dtype:Owl_symbolic_types.number_type -> int array -> Owl_symbolic_graph.symbolval max_pool2d : ?name:string -> ?padding:Owl_symbolic_types.pad -> int array -> int array -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval avg_pool2d : ?name:string -> ?padding:Owl_symbolic_types.pad -> int array -> int array -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval global_max_pool2d : ?name:string -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval global_avg_pool2d : ?name:string -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval dropout : ?name:string -> float -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval lambda : (Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbol) -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval fully_connected : ?name:string -> ?init_typ:Owl_symbolic_types.nn_init -> int -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval conv2d : ?name:string -> ?padding:Owl_symbolic_types.pad -> ?init_typ:Owl_symbolic_types.nn_init -> int array -> int array -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval transpose_conv2d : ?name:string -> ?padding:Owl_symbolic_types.pad -> ?init_typ:Owl_symbolic_types.nn_init -> int array -> int array -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval linear : ?name:string -> ?init_typ:Owl_symbolic_types.nn_init -> int -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval normalisation : ?name:string -> ?_axis:'a -> ?eps:float -> ?momentum:float -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval zero_padding2d : ?name:string -> int array -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval concat : ?name:string -> ?axis:int -> Owl_symbolic_graph.symbol array -> Owl_symbolic_graph.symbolval add : ?name:string -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval flt : ?name:string -> ?dtype:Owl_symbolic_types.number_type -> float -> Owl_symbolic_graph.symbolval tanh : ?name:string -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.symbolval get_network : ?name:string -> Owl_symbolic_graph.symbol -> Owl_symbolic_graph.t