package nx

You can search for identifiers within the package.

in-package search v0.2.0

package nx

nx
- Get started
- Input/Output Operations
- Library npy
  - Npy
    
    Batch_writer
    
    Header
    
    Npz
    
    Eq
    
    Kind
    
    Layout
- Library nx
  - Nx
- Library nx.c
  - Nx_c
- Library nx.core
  - Nx_core
    
    Dtype
    
    Shape
    
    View
    
    Backend_intf
    
    S
    
    Make_frontend
    
    B
    
    IntSet
    
    Error
- Library nx.io
  - Nx_io
- Library nx.native
  - Nx_native
- Library stb_image
  - Stb_image
- Library stb_image_write
  - Stb_image_write
- Sources
  - npy
    
    npy.ml
  - nx
    
    nx.ml
  - nx.c
    
    nx_c.ml
  - nx.core
    
    backend_intf.ml
    
    dtype.ml
    
    error.ml
    
    frontend.ml
    
    nx_core.ml
    
    nx_core__.ml
    
    shape.ml
    
    view.ml
  - nx.io
    
    nx_io.ml
    
    nx_io__.ml
    
    utils.ml
  - nx.native
    
    internal.ml
    
    nx_native.ml
    
    nx_native__.ml
    
    ops_binary.ml
    
    ops_cast.ml
    
    ops_im2col.ml
    
    ops_matmul.ml
    
    ops_reduce.ml
    
    ops_ternary.ml
    
    ops_threefry.ml
    
    ops_unary.ml
    
    parallel.ml
  - stb_image
    
    stb_image.ml
  - stb_image_write
    
    stb_image_write.ml

Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module `Nx_native`Source

Sourcetype ('a, 'b) buffer = ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t

Sourcetype context

Sourcetype ('a, 'b) t = {

context : context;
dtype : ('a, 'b) Nx_core.Dtype.t;
buffer : ('a, 'b) buffer;
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 data : ('a, 'b) t -> ('a, 'b) buffer

Sourceval context : ('a, 'b) t -> context

Sourceval with_view : ('a, 'b) t -> Nx_core.View.t -> ('a, 'b) t

Sourceval create_context : unit -> Nx_native__.Internal.context

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

Source

val op_const_array : 
  context ->
  ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t ->
  ('a, 'b) t

Sourceval op_add : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_mul : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_idiv : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_fdiv : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_max : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_mod : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_pow : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Source

val op_cmplt : 
  ('a, 'b) t ->
  ('a, 'b) Nx_native__.Internal.t ->
  (int, Nx_core.Dtype.uint8_elt) t

Source

val op_cmpne : 
  ('a, 'b) t ->
  ('a, 'b) Nx_native__.Internal.t ->
  (int, Nx_core.Dtype.uint8_elt) t

Sourceval op_xor : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_or : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_and : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t

Sourceval op_neg : ('a, 'b) t -> ('a, 'b) t

Sourceval op_log2 : ('a, 'b) t -> ('a, 'b) t

Sourceval op_exp2 : ('a, 'b) t -> ('a, 'b) t

Sourceval op_sin : ('a, 'b) t -> ('a, 'b) t

Sourceval op_sqrt : ('a, 'b) t -> ('a, 'b) t

Sourceval op_recip : ('a, 'b) t -> ('a, 'b) t

Source

val op_where : 
  (int, Nx_core.Dtype.uint8_elt) t ->
  ('a, 'b) t ->
  ('a, 'b) Nx_native__.Internal.t ->
  ('a, 'b) t

Source

val fill_buffer_with_identity : 
  ('a, 'b, 'c) Bigarray.Array1.t ->
  int ->
  'a ->
  unit

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

Source

val op_reduce_prod : 
  axes:int array ->
  keepdims:bool ->
  ('a, 'b) t ->
  ('a, 'b) t

Sourceval op_reshape : ('a, 'b) t -> int array -> ('a, 'b) t

Sourceval op_expand : ('a, 'b) t -> int array -> ('a, 'b) t

Sourceval op_permute : ('a, 'b) t -> int array -> ('a, 'b) t

Sourceval op_pad : ('a, 'b) t -> (int * int) array -> 'a -> ('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_cat : ('a, 'b) Nx_native__.Internal.t list -> int -> ('a, 'b) t

Sourceval op_cast : ('a, 'b) t -> ('c, 'd) Nx_core.Dtype.t -> ('c, 'd) t

Source

val op_contiguous : 
  ('a, 'b) Nx_native__.Internal.t ->
  ('a, 'b) Nx_native__.Internal.t

Source

val op_copy : 
  ('a, 'b) Nx_native__.Internal.t ->
  ('a, 'b) Nx_native__.Internal.t

Source

val op_assign : 
  ('a, 'b) Nx_native__.Internal.t ->
  ('a, 'b) Nx_native__.Internal.t ->
  unit

Source

val op_threefry : 
  (int32, Nx_core.Dtype.int32_elt) t ->
  (int32, Nx_core.Dtype.int32_elt) Nx_native__.Internal.t ->
  (int32, Nx_core.Dtype.int32_elt) t

Source

val op_gather : 
  ('a, 'b) t ->
  (int32, 'c) Nx_native__.Internal.t ->
  int ->
  ('a, 'b) t

Source

val op_scatter : 
  ?mode:[< `Add | `Set Set ] ->
  ?unique_indices:bool ->
  ('a, 'b) Nx_native__.Internal.t ->
  (int32, Nx_core.Dtype.int32_elt) Nx_native__.Internal.t ->
  ('a, 'b) Nx_native__.Internal.t ->
  int ->
  ('a, 'b) Nx_native__.Internal.t

Source

val op_unfold : 
  ('a, 'b) t ->
  kernel_size:int array ->
  stride:int array ->
  dilation:int array ->
  padding:(int * int) array ->
  ('a, 'b) Nx_native__.Internal.t

Source

val 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) Nx_native__.Internal.t

Source

val op_matmul : 
  ('a, 'b) t ->
  ('a, 'b) Nx_native__.Internal.t ->
  ('a, 'b) Nx_native__.Internal.t

package nx

Module Nx_nativeSource

Module `Nx_native`Source