package neural_nets_lib

  1. Overview
  2. Docs
include module type of struct include Initial_NTDSL end
val number : ?label:Base.string Base.list -> ?axis_label:Base.string -> Base.float -> Tensor.t
val ndarray : ?label:Base.string Base.list -> ?batch_dims:Base.int Base.list -> ?input_dims:Base.int Base.list -> ?output_dims:Base.int Base.list -> ?batch_axes:(Base.string * Base.int) Base.list -> ?input_axes:(Base.string * Base.int) Base.list -> ?output_axes:(Base.string * Base.int) Base.list -> ?strict:Base.bool -> Base.float Base.array -> Tensor.t
module O = NDO
val einsum : ?label:Base.string list -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val outer_sum : ?label:Base.string list -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val einsum1 : ?label:Base.string list -> Base.string -> Tensor.t -> Tensor.t
val term : label:Base.string Base.list -> ?batch_dims:Base.int Base.list -> ?input_dims:Base.int Base.list -> ?output_dims:Base.int Base.list -> ?batch_axes:(Base.string * Base.int) Base.list -> ?input_axes:(Base.string * Base.int) Base.list -> ?output_axes:(Base.string * Base.int) Base.list -> ?deduced:Shape.deduce_within_shape -> ?init_op:Tensor.init_op -> ?fetch_op:(v:Tensor.tn -> Tensor.fetch_op) -> Base.unit -> Tensor.t
val range : ?label:Base.string list -> ?axis_label:Base.string -> Base.Int.t -> Tensor.t
val range_of_shape : ?label:Base.string list -> ?batch_dims:Base.Int.t Base.List.t -> ?input_dims:Base.Int.t Base.List.t -> ?output_dims:Base.Int.t Base.List.t -> ?batch_axes:(Base.string * Base.Int.t) Base.List.t -> ?input_axes:(Base.string * Base.Int.t) Base.List.t -> ?output_axes:(Base.string * Base.Int.t) Base.List.t -> unit -> Tensor.t
val counter : ?label:Base.string list -> Tensor.t -> Tensor.t
OCaml

Innovation. Community. Security.