package tensorflow

  1. Overview
  2. Docs
module Input_name : sig ... end
type _1d
type _2d
type _3d
module Shape : sig ... end
type ('a, 'b) t
type init = [
  1. | `const of float
  2. | `normal of float
  3. | `truncated_normal of float
]
val shape : ('a, 'b) t -> 'a Shape.t
val default_input : ('a, 'b) t -> 'b Input_name.t option
val node : ('a, 'b) t -> 'b Node.t
val type_ : ('a, 'b) t -> 'b Node.Type.t
val input : shape:'a Shape.t -> type_:'b Node.Type.t -> ('a, 'b) t
val named_input : shape:'a Shape.t -> type_:'b Node.Type.t -> 'b Input_name.t * ('a, 'b) t
val dense : ?w_init:init -> ?b_init:init -> (_1d, [< `double | `float ] as 'b) t -> shape:int -> (_1d, 'b) t
val conv2d : ?w_init:init -> ?b_init:init -> (_3d, [< `double | `float ] as 'b) t -> filter:(int * int) -> out_channels:int -> strides:(int * int) -> padding:[ `same | `valid ] -> (_3d, 'b) t
val sigmoid : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t
val tanh : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t
val relu : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t
val softmax : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t
val max_pool : (_3d, [ `float ]) t -> filter:(int * int) -> strides:(int * int) -> padding:[ `same | `valid ] -> (_3d, [ `float ]) t
val concat : (_1d, 'a) t -> (_1d, 'a) t -> (_1d, 'a) t
val (+) : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t -> ('a, 'c) t
val (-) : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t -> ('a, 'c) t
val (*) : ('a, [< `double | `float ] as 'c) t -> ('a, 'c) t -> ('a, 'c) t
val f : float -> shape:'a Shape.t -> ('a, [ `float ]) t
val reshape : ('a, 'c) t -> shape:'b Shape.t -> ('b, 'c) t
val flatten : ('a, 'b) t -> (_1d, 'b) t
val split : (_2d, 'a) t -> (_1d, 'a) t list
val concatN : (_1d, 'a) t list -> (_2d, 'a) t
module Shared_var : sig ... end
OCaml

Innovation. Community. Security.