package tensorflow

  1. Overview
  2. Docs
type _1d
type _2d
type _3d
module Shape : sig ... end
module Input_id : sig ... end
type 'a t
type init = [
  1. | `const of float
  2. | `normal of float
  3. | `truncated_normal of float
]
val shape : 'a t -> 'a Shape.t
val input : shape:'a Shape.t -> 'a t * Input_id.t
val const : float -> shape:'a Shape.t -> 'a t
val sigmoid : 'a t -> 'a t
val tanh : 'a t -> 'a t
val relu : 'a t -> 'a t
val softmax : 'a t -> 'a t
val (+) : 'a t -> 'a t -> 'a t
val (-) : 'a t -> 'a t -> 'a t
val (*) : 'a t -> 'a t -> 'a t
val dense : ?w_init:init -> ?b_init:init -> int -> _1d t -> _1d t
val conv2d : ?w_init:init -> ?b_init:init -> filter:(int * int) -> out_channels:int -> strides:(int * int) -> padding:[ `same | `valid ] -> unit -> _3d t -> _3d t
val dense' : ?w_init:init -> ?b_init:init -> int -> (_1d t -> _1d t) Core_kernel.Std.Staged.t
val conv2d' : ?w_init:init -> ?b_init:init -> filter:(int * int) -> out_channels:int -> strides:(int * int) -> padding:[ `same | `valid ] -> unit -> (_3d t -> _3d t) Core_kernel.Std.Staged.t
val max_pool : _3d t -> filter:(int * int) -> strides:(int * int) -> padding:[ `same | `valid ] -> _3d t
val reshape : 'b t -> shape:'a Shape.t -> 'a t
val flatten : 'a t -> _1d t
val split : _2d t -> _1d t list
val concat : _1d t list -> _2d t
module Optimizer : sig ... end
module Loss : sig ... end
module Model : sig ... end
OCaml

Innovation. Community. Security.