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