tensorflow 0.0.6 · OCaml Package

module Placeholder : sig ... end

type 't b = ?name:string -> 't Node.t -> 't Node.t -> 't Node.t

val (-) : [< `complex64 | `double | `float | `int32 | `int64 ] b

val (/) : [< `complex64 | `double | `float | `int32 | `int64 ] b

val (*) : [< `complex64 | `double | `float | `int32 | `int64 ] b

val (*^) : [< `complex64 | `double | `float | `int32 ] b

val f_or_d : 
  ?shape:int list ->
  type_:[< `double | `float ] as 'b Node.Type.t ->
  float ->
  'b Node.t

val f : ?shape:int list -> float -> [ `float ] Node.t

val d : ?shape:int list -> float -> [ `double ] Node.t

val cf : ?shape:int list -> float list -> [ `float ] Node.t

val cd : ?shape:int list -> float list -> [ `double ] Node.t

val const_float : 
  ?name:string ->
  ?control_inputs:Node.p list ->
  ?shape:int list ->
  type_:[< `double | `float ] as 'a Node.Type.t ->
  float list ->
  'a Node.t

val const_int : 
  ?name:string ->
  ?control_inputs:Node.p list ->
  ?shape:int list ->
  type_:[< `int32 | `int64 ] as 'a Node.Type.t ->
  int list ->
  'a Node.t

val const_string : 
  ?name:string ->
  ?shape:int list ->
  string list ->
  [ `string ] Node.t

val scalar : 
  ?empty_shape:unit ->
  type_:[< `double | `float ] as 'a Node.Type.t ->
  float ->
  'a Node.t

val four32 : [ `int32 ] Node.t

val three32 : [ `int32 ] Node.t

val two32 : [ `int32 ] Node.t

val one32 : [ `int32 ] Node.t

val zero32 : [ `int32 ] Node.t

type 'b reduce_fn = ?dims:int list -> 'b Node.t -> 'b Node.t constraint 'b = [< `complex64 | `double | `float | `int32 | `int64 ]

val reduce_sum : [< `complex64 | `double | `float | `int32 | `int64 ] reduce_fn

val reduce_min : [< `complex64 | `double | `float | `int32 | `int64 ] reduce_fn

val reduce_max : [< `complex64 | `double | `float | `int32 | `int64 ] reduce_fn

val reduce_mean : 
  [< `complex64 | `double | `float | `int32 | `int64 ] reduce_fn

val reduce_prod : 
  [< `complex64 | `double | `float | `int32 | `int64 ] reduce_fn

val reduce_all : ?dims:int list -> [ `bool ] Node.t -> [ `bool ] Node.t

val reduce_any : ?dims:int list -> [ `bool ] Node.t -> [ `bool ] Node.t

val save_ : 
  ?name:string ->
  [ `string ] Node.t ->
  [ `string ] Node.t ->
  Node.p list ->
  [ `unit ] Node.t

val save : filename:string -> (string * Node.p) list -> [ `unit ] Node.t

val split2 : 
  ?name:string ->
  [ `int32 ] Node.t ->
  't Node.t ->
  't Node.t * 't Node.t

val split3 : 
  ?name:string ->
  [ `int32 ] Node.t ->
  't Node.t ->
  't Node.t * 't Node.t * 't Node.t

val split4 : 
  ?name:string ->
  [ `int32 ] Node.t ->
  't Node.t ->
  't Node.t * 't Node.t * 't Node.t * 't Node.t

val range : [ `int32 ] Node.t -> [ `int32 ] Node.t

val placeholder : 
  ?name:string ->
  type_:'a Node.Type.t ->
  int list ->
  'a Placeholder.t

val dropout : 
  [< `double | `float ] as 'b Node.t ->
  keep_prob:'b Node.t ->
  'b Node.t

val cast : 
  ?name:string ->
  'srcT Node.t ->
  type_:'dstT Node.Type.t ->
  'dstT Node.t

val count : 'a Node.t -> dims:int list -> [ `int32 ] Node.t

type !'a moments = {

mean : 'a Node.t;
variance : 'a Node.t;

}

val moments : [< `double | `float ] as 'b Node.t -> dims:int list -> 'b moments

val normalize : 
  ?epsilon:float ->
  [< `double | `float ] as 'b Node.t ->
  'b moments ->
  'b Node.t

val cond_with_control_inputs : 
  [ `bool ] Node.t ->
  if_true:(control_inputs:Node.p list -> 'a Node.t) ->
  if_false:(control_inputs:Node.p list -> 'a Node.t) ->
  'a Node.t

val cond : 
  [ `bool ] Node.t ->
  if_true:'a Node.t ->
  if_false:'a Node.t ->
  'a Node.t

package tensorflow