package nx

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

Install

dune-project
 Dependency

Authors

Maintainers

Sources

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

doc/nx.core/Nx_core/index.html

Module Nx_coreSource

Core modules for nx.

This module re-exports core building blocks used by backends and the high-level Nx frontend.

Sourcemodule Dtype : sig ... end

Tensor element dtypes.

Sourcemodule Shape : sig ... end

Concrete shape operations.

Sourcemodule View : sig ... end

Strided tensor views.

Sourcemodule Backend_intf : sig ... end

Backend interface used by frontend functors.

Sourcemodule Rng : sig ... end

RNG key utilities.

Sourcemodule Make_frontend (B : Backend_intf.S) : sig ... end

Frontend functor parameterized by a backend implementation.