tensorflow

TensorFlow bindings for OCaml
IN THIS PACKAGE
Module Tensorflow . Node
module Session : sig ... end
module Op_name : Base.Identifiable.S
module Name : Base.Identifiable.S
module Id : Base.Identifiable.S
type 'a t
type p =
| P : _ t -> p
type input = [
| `single of p
| `multi of p Base.list
]
val create : name:Name.t -> op_name:Op_name.t -> output_type:'a Type.t -> inputs:input Base.list -> control_inputs:p Base.list -> attributes:(Base.string * attr) Base.list -> output_idx:Base.int Base.option -> 'a t
val create_gradient : output -> output_type:'a Type.t -> 'a t
val name : _ t -> Name.t
val op_name : _ t -> Op_name.t
val output_type : 'a t -> 'a Type.t
val inputs : _ t -> input Base.list
val flat_inputs : _ t -> p Base.list
val control_inputs : _ t -> p Base.list
val attributes : _ t -> (Base.string * attr) Base.list
val output_idx : _ t -> Base.int Base.option
val operation : _ t -> Tensorflow_core.Operation.t
val output : _ t -> output
val packed_name : p -> Name.t
val packed_op_name : p -> Op_name.t
val packed_inputs : p -> input Base.list
val packed_flat_inputs : p -> p Base.list
val packed_is_real : p -> Base.bool
val packed_id : p -> Id.t
val packed_output_idx : p -> Base.int Base.option
val packed_operation : p -> Tensorflow_core.Operation.t
val packed_output : p -> output
val get_attr_bool : _ t -> Base.string -> Base.bool Base.option
val get_attr_string : _ t -> Base.string -> Base.string Base.option
val get_attr_int : _ t -> Base.string -> Base.int Base.option
val get_attr_int_list : _ t -> Base.string -> Base.int Base.list Base.option
val shape : _ t -> Base.int Base.list
val set_output_idx : 'a t -> Base.int Base.option -> 'a t
val set_output_idx_and_output_type : 'b t -> Base.int Base.option -> type_:'a Type.t -> 'a t
val id : _ t -> Id.t
val extract : p -> 'a Type.t -> 'a t Base.option
val extract_exn : p -> 'a Type.t -> 'a t