Legend:
Library
Module
Module type
Parameter
Class
Class type
Library
Module
Module type
Parameter
Class
Class type
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