package nx

  1. Overview
  2. Docs
package nx
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Nx_cSource

Sourceval assign : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval copy : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval cast : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('d, 'e, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval neg : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval sqrt : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval sin : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval exp2 : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval log2 : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval recip : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval add : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval sub : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval mul : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval fdiv : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval max : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval mod_ : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval pow : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval idiv : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval xor : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval or_ : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval and_ : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval cmplt : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> (int, Bigarray.int8_unsigned_elt, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval cmpne : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> (int, Bigarray.int8_unsigned_elt, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval reduce_sum : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> int -> unit
Sourceval reduce_max : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> int -> unit
Sourceval reduce_prod : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> int -> unit
Sourceval where : int -> int array -> (int, Bigarray.int8_unsigned_elt, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval pad : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> 'a -> unit
Sourceval cat : (('a, 'b, 'c) Bigarray.Array1.t * Nx_core.View.t) array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> unit
Sourceval threefry : int -> int array -> (int32, Bigarray.int32_elt, 'c) Bigarray.Array1.t -> int array -> int -> (int32, Bigarray.int32_elt, 'c) Bigarray.Array1.t -> int array -> int -> (int32, Bigarray.int32_elt, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval gather : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> (int32, Bigarray.int32_elt, 'c) Bigarray.Array1.t -> int array -> int -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> unit
Sourceval scatter : int -> int array -> int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> (int32, Bigarray.int32_elt, 'c) Bigarray.Array1.t -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int -> unit
Sourcetype ('a, 'b) buffer = ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t
Sourcetype context = unit
Sourceval create_context : unit -> unit
Sourcetype ('a, 'b) t = {
  1. context : context;
  2. dtype : ('a, 'b) Nx_core.Dtype.t;
  3. buffer : ('a, 'b) buffer;
  4. view : Nx_core.View.t;
}
Sourceval view : ('a, 'b) t -> Nx_core.View.t
Sourceval dtype : ('a, 'b) t -> ('a, 'b) Nx_core.Dtype.t
Sourceval context : ('a, 'b) t -> context
Sourceval data : ('a, 'b) t -> ('a, 'b) buffer
Sourceval create : context -> ('a, 'b) Nx_core.Dtype.t -> ('a, 'b) buffer -> Nx_core.View.t -> ('a, 'b) t
Sourceval make_buffer : ('a, 'b) Nx_core.Dtype.t -> int -> ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t
Sourceval make_tensor : ('a, 'b) t -> Nx_core.Shape.t -> ('a, 'b) t
Sourceval op_buffer : context -> ('a, 'b) Nx_core.Dtype.t -> int -> ('a, 'b) t
Sourceval op_const_scalar : context -> 'a -> ('a, 'b) Nx_core.Dtype.t -> ('a, 'b) t
Sourceval op_const_array : context -> ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t -> ('a, 'b) t
Sourceval op_copy : ('a, 'b) t -> ('a, 'b) t
Sourceval op_contiguous : ('a, 'b) t -> ('a, 'b) t
Sourceval op_assign : ('a, 'b) t -> ('a, 'b) t -> unit
Sourceval op_cast : ('a, 'b) t -> ('c, 'd) Nx_core.Dtype.t -> ('c, 'd) t
Sourceval unop : (int -> Nx_core.Shape.t -> ('a, 'b) buffer -> int array -> int -> ('a, 'b) buffer -> int array -> int -> 'c) -> ('a, 'b) t -> ('a, 'b) t
Sourceval binop : (int -> Nx_core.Shape.t -> ('a, 'b) buffer -> int array -> int -> ('c, 'd) buffer -> int array -> int -> ('a, 'b) buffer -> int array -> int -> 'e) -> ('a, 'b) t -> ('c, 'd) t -> ('a, 'b) t
Sourceval op_neg : ('a, 'b) t -> ('a, 'b) t
Sourceval op_sqrt : ('a, 'b) t -> ('a, 'b) t
Sourceval op_sin : ('a, 'b) t -> ('a, 'b) t
Sourceval op_exp2 : ('a, 'b) t -> ('a, 'b) t
Sourceval op_log2 : ('a, 'b) t -> ('a, 'b) t
Sourceval op_recip : ('a, 'b) t -> ('a, 'b) t
Sourceval op_add : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_sub : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_mul : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_fdiv : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_max : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_mod : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_pow : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_idiv : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_xor : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_or : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_and : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval binop_cmp : (int -> Nx_core.Shape.t -> ('a, 'b) buffer -> int array -> int -> ('c, 'd) buffer -> int array -> int -> (int, Nx_core.Dtype.uint8_elt) buffer -> int array -> int -> 'e) -> ('a, 'b) t -> ('c, 'd) t -> (int, Nx_core.Dtype.uint8_elt) t
Sourceval op_cmplt : ('a, 'b) t -> ('a, 'b) t -> (int, Nx_core.Dtype.uint8_elt) t
Sourceval op_cmpne : ('a, 'b) t -> ('a, 'b) t -> (int, Nx_core.Dtype.uint8_elt) t
Sourceval reduce_op : (int -> Nx_core.Shape.t -> ('a, 'b) buffer -> int array -> int -> ('a, 'b) buffer -> int array -> int -> int array -> int -> 'c) -> axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reduce_sum : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reduce_max : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reduce_prod : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reshape : ('a, 'b) t -> Nx_core.Shape.t -> ('a, 'b) t
Sourceval op_expand : ('a, 'b) t -> Nx_core.Shape.t -> ('a, 'b) t
Sourceval op_permute : ('a, 'b) t -> int array -> ('a, 'b) t
Sourceval op_shrink : ('a, 'b) t -> (int * int) array -> ('a, 'b) t
Sourceval op_flip : ('a, 'b) t -> bool array -> ('a, 'b) t
Sourceval op_where : (int, Bigarray.int8_unsigned_elt) t -> ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_cat : ('a, 'b) t list -> int -> ('a, 'b) t
Sourceval op_threefry : (int32, Bigarray.int32_elt) t -> (int32, Bigarray.int32_elt) t -> (int32, Bigarray.int32_elt) t
Sourceval op_gather : ('a, 'b) t -> (int32, Bigarray.int32_elt) t -> int -> ('a, 'b) t
Sourceval op_scatter : ?mode:[< `Add | `Set Set ] -> ?unique_indices:bool -> ('a, 'b) t -> (int32, Bigarray.int32_elt) t -> ('a, 'b) t -> int -> ('a, 'b) t
Sourceval matmul : ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int array -> int -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int array -> int -> unit
Sourceval unfold : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> int array -> int array -> int array -> int array -> unit
Sourceval fold : int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int -> int array -> ('a, 'b, 'c) Bigarray.Array1.t -> int array -> int -> int array -> int array -> int array -> int array -> int array -> unit
Sourceval op_matmul : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_unfold : ('a, 'b) t -> kernel_size:int array -> stride:int array -> dilation:int array -> padding:(int * int) array -> ('a, 'b) t
Sourceval op_fold : ('a, 'b) t -> output_size:int array -> kernel_size:int array -> stride:int array -> dilation:int array -> padding:(int * int) array -> ('a, 'b) t
Sourceval op_pad : ('a, 'b) t -> (int * int) array -> 'a -> ('a, 'b) t
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct