Legend:
Library
Module
Module type
Parameter
Class
Class type
type number_type =
  1. | SNT_Noop
  2. | SNT_Float
  3. | SNT_Double
  4. | SNT_Complex32
  5. | SNT_Complex64
  6. | SNT_Bool
  7. | SNT_String
  8. | SNT_Int8
  9. | SNT_Int16
  10. | SNT_Int32
  11. | SNT_Int64
  12. | SNT_Uint8
  13. | SNT_Uint16
  14. | SNT_Uint32
  15. | SNT_Uint64
  16. | SNT_Float16
  17. | SNT_SEQ of number_type
val number_type_to_string : number_type -> string
type tensor = {
  1. mutable dtype : number_type;
  2. mutable shape : int array;
  3. mutable str_val : string array option;
  4. mutable flt_val : float array option;
  5. mutable int_val : int array option;
  6. mutable raw_val : bytes option;
}
type pad =
  1. | SAME_UPPER
  2. | SAME_LOWER
  3. | VALID
  4. | PAD of int array
val make_tensor : ?dtype:number_type -> ?flt_val:float array -> ?int_val:int array -> ?str_val:string array -> ?raw_val:bytes -> int array -> tensor
val get_tensor_dtype : tensor -> number_type
type activation =
  1. | Relu
  2. | Tanh
  3. | Sigmoid
  4. | Affine of float * float
  5. | LeakyRelu of float
  6. | ThresholdedRelu of float
  7. | ScaledTanh of float * float
  8. | HardSigmoid of float * float
  9. | Elu of float
  10. | Softsign
  11. | Softplus
  12. | Softmax of int
val activation_to_string : activation -> string
type nn_init =
  1. | Uniform of float * float
  2. | Gaussian of float * float
  3. | Standard
  4. | Tanh
type attrvalue =
  1. | ATTR_Nil
  2. | ATTR_Int of int
  3. | ATTR_Bool of bool
  4. | ATTR_Type of number_type
  5. | ATTR_Float of float
  6. | ATTR_Shape of int array
  7. | ATTR_String of string
  8. | ATTR_Tensor of tensor
  9. | ATTR_Array of attrvalue array
  10. | ATTR_NameArray of {
    1. name : string;
    2. attr : (string * attrvalue) array;
    }
val get_attrvalue_int : attrvalue -> int
val get_attrvalue_float : attrvalue -> float
val get_attrvalue_type : attrvalue -> number_type
val get_attrvalue_shape : attrvalue -> int array
exception TYPE_CHECK of string
exception INVALID_NAME of string