package nx

  1. Overview
  2. Docs
N-dimensional arrays for OCaml

Install 

dune-project
 Dependency

github.com Readme Changelog Edit opam file Versions (4)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

raven-1.0.0.alpha3.tbz
sha256=96d35ce03dfbebd2313657273e24c2e2d20f9e6c7825b8518b69bd1d6ed5870f
sha512=90c5053731d4108f37c19430e45456063e872b04b8a1bbad064c356e1b18e69222de8bfcf4ec14757e71f18164ec6e4630ba770dbcb1291665de5418827d1465

doc/nx.core/Nx_core/Make_frontend/Einsum/index.html

Module Make_frontend.EinsumSource

Sourcetype token =
  1. | Axis of char
  2. | Ellipsis
Sourceval parse_operand : string -> token list
Sourceval parse_equation : string -> token list array * token list option
Sourceval handle_repeated_indices : ('a, 'b) B.t -> token list -> ('a, 'b) B.t * token list
Sourcetype tensor_info = {
  1. id : int;
  2. shape : int array;
  3. axis_labels : char list;
}
Sourcetype contraction_path =
  1. | Leaf of int
  2. | Node of contraction_path * contraction_path * tensor_info
Sourceval estimate_cost : tensor_info -> tensor_info -> Char.t list -> float * float
Sourceval optimize_path : tensor_info list -> Char.t list -> contraction_path
Sourceval contract_pair : ('a, 'b) B.t -> String.t -> ('a, 'b) B.t -> String.t -> String.t -> ('a, 'b) B.t
Sourceval calculate : string -> ('a, 'b) B.t array -> ('a, 'b) B.t