package nx

You can search for identifiers within the package.

in-package search v0.2.0

On This Page

Parameters
Signature

package nx

nx
- CHANGES
- README
- Library nx
  - Nx
    
    Rng
    
    Infix
- Library nx.backend
  - Nx_backend
- Library nx.buffer
  - Nx_buffer
- Library nx.c
  - Nx_backend
- Library nx.core
  - Nx_core
    
    Dtype
    
    Packed
    
    Shape
    
    View
    
    Backend_intf
    
    S
    
    Rng
    
    Make_frontend
    
    B
    
    IntSet
    
    Einsum
    
    Infix
- Library nx.effect
  - Nx_effect
- Library nx.io
  - Nx_io
- Library nx.pocketfft
  - Pocketfft
- Library nx.top
  - Nx_top
- Library nx.zip
  - Gzip
  - Zip
  - Zlib
- Library stb_image
  - Stb_image
- Library stb_image_write
  - Stb_image_write
- Sources
  - nx
    
    nx.ml
  - nx.buffer
    
    nx_buffer.ml
  - nx.c
    
    nx_backend.ml
    
    nx_backend__nx_c__.ml
  - nx.core
    
    backend_intf.ml
    
    dtype.ml
    
    frontend.ml
    
    nx_core.ml
    
    nx_core__.ml
    
    rng.ml
    
    shape.ml
    
    view.ml
  - nx.effect
    
    nx_effect.ml
  - nx.io
    
    error.ml
    
    npy.ml
    
    nx_io.ml
    
    nx_io__.ml
    
    nx_npy.ml
    
    nx_safetensors.ml
    
    nx_txt.ml
    
    packed_nx.ml
    
    safetensors.ml
  - nx.pocketfft
    
    pocketfft.ml
  - nx.top
    
    nx_top.ml
  - nx.zip
    
    gzip.ml
    
    zip.ml
    
    zlib.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_core.Make_frontend`Source

Frontend functor parameterized by a backend implementation.

Parameters

module B : Backend_intf.S

Signature

Sourcemodule B = B

Sourceval err : string -> ('a, unit, string, 'b) format4 -> 'a

Sourcetype ('a, 'b) t = ('a, 'b) B.t

Sourcetype context = B.context

Sourcetype float16_elt = Nx_buffer.float16_elt

Sourcetype float32_elt = Nx_buffer.float32_elt

Sourcetype float64_elt = Nx_buffer.float64_elt

Sourcetype bfloat16_elt = Nx_buffer.bfloat16_elt

Sourcetype float8_e4m3_elt = Nx_buffer.float8_e4m3_elt

Sourcetype float8_e5m2_elt = Nx_buffer.float8_e5m2_elt

Sourcetype int4_elt = Nx_buffer.int4_signed_elt

Sourcetype uint4_elt = Nx_buffer.int4_unsigned_elt

Sourcetype int8_elt = Nx_buffer.int8_signed_elt

Sourcetype uint8_elt = Nx_buffer.int8_unsigned_elt

Sourcetype int16_elt = Nx_buffer.int16_signed_elt

Sourcetype uint16_elt = Nx_buffer.int16_unsigned_elt

Sourcetype int32_elt = Nx_buffer.int32_elt

Sourcetype uint32_elt = Nx_buffer.uint32_elt

Sourcetype int64_elt = Nx_buffer.int64_elt

Sourcetype uint64_elt = Nx_buffer.uint64_elt

Sourcetype complex32_elt = Nx_buffer.complex32_elt

Sourcetype complex64_elt = Nx_buffer.complex64_elt

Sourcetype bool_elt = Nx_buffer.bool_elt

Sourcetype ('a, 'b) dtype = ('a, 'b) Dtype.t =

| Float16 : (float, float16_elt) dtype
| Float32 : (float, float32_elt) dtype
| Float64 : (float, float64_elt) dtype
| BFloat16 : (float, bfloat16_elt) dtype
| Float8_e4m3 : (float, float8_e4m3_elt) dtype
| Float8_e5m2 : (float, float8_e5m2_elt) dtype
| Int4 : (int, int4_elt) dtype
| UInt4 : (int, uint4_elt) dtype
| Int8 : (int, int8_elt) dtype
| UInt8 : (int, uint8_elt) dtype
| Int16 : (int, int16_elt) dtype
| UInt16 : (int, uint16_elt) dtype
| Int32 : (int32, int32_elt) dtype
| UInt32 : (int32, uint32_elt) dtype
| Int64 : (int64, int64_elt) dtype
| UInt64 : (int64, uint64_elt) dtype
| Complex64 : (Complex.t, complex32_elt) dtype
| Complex128 : (Complex.t, complex64_elt) dtype
| Bool : (bool, bool_elt) dtype

Sourcetype float16_t = (float, float16_elt) t

Sourcetype float32_t = (float, float32_elt) t

Sourcetype float64_t = (float, float64_elt) t

Sourcetype int8_t = (int, int8_elt) t

Sourcetype uint8_t = (int, uint8_elt) t

Sourcetype int16_t = (int, int16_elt) t

Sourcetype uint16_t = (int, uint16_elt) t

Sourcetype int32_t = (int32, int32_elt) t

Sourcetype int64_t = (int64, int64_elt) t

Sourcetype uint32_t = (int32, uint32_elt) t

Sourcetype uint64_t = (int64, uint64_elt) t

Sourcetype complex64_t = (Complex.t, complex32_elt) t

Sourcetype complex128_t = (Complex.t, complex64_elt) t

Sourcetype bool_t = (bool, bool_elt) t

Sourceval float16 : (float, float16_elt) dtype

Sourceval float32 : (float, float32_elt) dtype

Sourceval float64 : (float, float64_elt) dtype

Sourceval bfloat16 : (float, bfloat16_elt) dtype

Sourceval float8_e4m3 : (float, float8_e4m3_elt) dtype

Sourceval float8_e5m2 : (float, float8_e5m2_elt) dtype

Sourceval int4 : (int, int4_elt) dtype

Sourceval uint4 : (int, uint4_elt) dtype

Sourceval int8 : (int, int8_elt) dtype

Sourceval uint8 : (int, uint8_elt) dtype

Sourceval int16 : (int, int16_elt) dtype

Sourceval uint16 : (int, uint16_elt) dtype

Sourceval int32 : (int32, int32_elt) dtype

Sourceval uint32 : (int32, uint32_elt) dtype

Sourceval int64 : (int64, int64_elt) dtype

Sourceval uint64 : (int64, uint64_elt) dtype

Sourceval complex64 : (Complex.t, complex32_elt) dtype

Sourceval complex128 : (Complex.t, complex64_elt) dtype

Sourceval bool : (bool, bool_elt) dtype

Sourcetype index =

| I of int
| L of int list
| R of int * int
| Rs of int * int * int
| A
| M of (bool, bool_elt) t
| N

Sourceval data : ('a, 'b) B.t -> ('a, 'b) Nx_buffer.t

Sourceval shape : ('a, 'b) B.t -> int array

Sourceval dtype : ('a, 'b) B.t -> ('a, 'b) Dtype.t

Sourceval itemsize : ('a, 'b) B.t -> int

Sourceval strides : ('a, 'b) B.t -> int array

Sourceval stride : int -> ('a, 'b) B.t -> int

Sourceval dims : ('a, 'b) B.t -> int array

Sourceval dim : int -> ('a, 'b) B.t -> int

Sourceval ndim : ('a, 'b) B.t -> int

Sourceval size : ('a, 'b) B.t -> int

Sourceval numel : ('a, 'b) B.t -> int

Sourceval nbytes : ('a, 'b) B.t -> int

Sourceval offset : ('a, 'b) B.t -> int

Sourceval is_c_contiguous : ('a, 'b) B.t -> bool

Sourceval array_prod : int array -> int

Sourcemodule IntSet : sig ... end

Sourceval power_of_two : 'a 'b. ('a, 'b) Dtype.t -> int -> 'a

Sourceval ensure_float_dtype : string -> ('a, 'b) B.t -> unit

Sourceval ensure_int_dtype : string -> ('a, 'b) B.t -> unit

Sourceval resolve_axis : ?ndim_opt:int -> ('a, 'b) B.t -> int option -> int array

Sourceval resolve_single_axis : ?ndim_opt:int -> ('a, 'b) B.t -> int -> int

Sourceval normalize_and_dedup_axes : op:string -> int -> int list -> int list

Sourceval reduction_element_count : int array -> ?axes:int list -> unit -> int

Sourceval copy_to_out : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval reshape : Shape.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval broadcast_shapes : Shape.t -> Shape.t -> int array

Sourceval broadcast_to : Shape.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val broadcasted : 
  ?reverse:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t * ('a, 'b) B.t

Sourceval expand : int array -> ('a, 'b) B.t -> ('a, 'b) B.t

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

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

Sourceval contiguous : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval copy : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval blit : ('a, 'b) B.t -> ('a, 'b) B.t -> unit

Source

val create : 
  B.context ->
  ('a, 'b) Dtype.t ->
  int array ->
  'a array ->
  ('a, 'b) B.t

Source

val init : 
  B.context ->
  ('a, 'b) Dtype.t ->
  Shape.t ->
  (int array -> 'a) ->
  ('a, 'b) B.t

Sourceval scalar : B.context -> ('a, 'b) Dtype.t -> 'a -> ('a, 'b) B.t

Sourceval scalar_like : ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval fill : 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval empty : B.context -> ('a, 'b) Dtype.t -> int array -> ('a, 'b) B.t

Sourceval zeros : B.context -> ('a, 'b) Dtype.t -> int array -> ('a, 'b) B.t

Sourceval ones : B.context -> ('a, 'b) Dtype.t -> int array -> ('a, 'b) B.t

Sourceval full : B.context -> ('a, 'b) Dtype.t -> int array -> 'a -> ('a, 'b) B.t

Source

val create_like : 
  ('a, 'b) B.t ->
  (B.context -> ('a, 'b) Dtype.t -> int array -> 'c) ->
  'c

Sourceval empty_like : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval full_like : ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval zeros_like : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval ones_like : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval to_buffer : ('a, 'b) B.t -> ('a, 'b) Nx_buffer.t

Source

val to_bigarray : 
  ('c, 'd) B.t ->
  ('a, 'b, Bigarray.c_layout) Bigarray.Genarray.t

Source

val of_buffer : 
  B.context ->
  shape:Shape.t ->
  ('a, 'b) Nx_buffer.t ->
  ('a, 'b) B.t

Sourceval of_bigarray : B.context -> 'c -> ('a, 'b) B.t

Sourceval to_array : ('a, 'b) B.t -> 'a array

Source

val binop : 
  ?out:('a, 'b) B.t ->
  (out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> 'c) ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val cmpop : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  (out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> 'c) ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Sourceval add : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval add_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval radd_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sub : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sub_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rsub_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval mul : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval mul_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rmul_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval div : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval div_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rdiv_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval pow : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval pow_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rpow_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval maximum : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval maximum_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rmaximum_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval minimum : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval minimum_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rminimum_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval mod_ : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval mod_s : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t

Sourceval rmod_s : ?out:('a, 'b) B.t -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val bitwise_xor : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val bitwise_or : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val bitwise_and : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val logical_and : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val logical_or : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val logical_xor : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval logical_not : ?out:('a, 'b) B.t -> ('a, 'b) t -> ('a, 'b) t

Source

val cmpeq : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val cmpne : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val cmplt : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val cmple : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val cmpgt : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val cmpge : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val less : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val less_equal : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val greater : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val greater_equal : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val equal : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val not_equal : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val equal_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val not_equal_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val less_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val greater_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val less_equal_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val greater_equal_s : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  'a ->
  (bool, Dtype.bool_elt) B.t

Source

val unaryop : 
  ?out:('a, 'b) B.t ->
  (out:('a, 'b) B.t -> ('a, 'b) B.t -> 'c) ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval neg : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval bitwise_not : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval invert : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sin : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cos : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sqrt : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval recip : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval log : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval exp : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval abs : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval log2 : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval exp2 : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval tan : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval square : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sign : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval relu : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sigmoid : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval rsqrt : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval asin : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval acos : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval atan : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval sinh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cosh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval tanh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval asinh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval acosh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval atanh : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval trunc : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval ceil : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval floor : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval round : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val isinf : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val isnan : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val isfinite : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val lerp : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val lerp_scalar_weight : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  'a ->
  ('a, 'b) B.t

Source

val shift_op : 
  op:string ->
  apply:(?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t) ->
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  int ->
  ('a, 'b) B.t

Sourceval lshift : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> int -> ('a, 'b) B.t

Sourceval rshift : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> int -> ('a, 'b) B.t

Source

val clamp : 
  ?out:('a, 'b) B.t ->
  ?min:'a ->
  ?max:'a ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val clip : 
  ?out:('a, 'b) B.t ->
  ?min:'a ->
  ?max:'a ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val where : 
  ?out:('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval atan2 : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval hypot : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval reduce_output_shape : int array -> int array -> bool -> int array

Source

val reduce_op : 
  ?out:('a, 'b) B.t ->
  (out:('a, 'b) B.t -> axes:int array -> keepdims:bool -> ('a, 'b) B.t -> 'c) ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val sum : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val max : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val min : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val prod : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val associative_scan : 
  axis:int ->
  [ `Max | `Min | `Prod | `Sum ] ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val cumulative_scan : 
  ?axis:int ->
  [ `Max | `Min | `Prod | `Sum ] ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval cumsum : ?axis:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cumprod : ?axis:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cummax : ?axis:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cummin : ?axis:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val mean : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val var : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ?ddof:int ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val std : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ?ddof:int ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val all : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Source

val any : 
  ?out:(bool, Dtype.bool_elt) B.t ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  (bool, Dtype.bool_elt) B.t

Sourceval array_equal : ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t

Sourceval pad : (int * int) array -> 'a -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval shrink : (int * int) array -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval flatten : ?start_dim:int -> ?end_dim:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval unflatten : int -> int array -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval ravel : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval squeeze : ?axes:IntSet.elt list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval unsqueeze : ?axes:IntSet.elt list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval squeeze_axis : IntSet.elt -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval unsqueeze_axis : IntSet.elt -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval expand_dims : IntSet.elt list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval transpose : ?axes:int list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval flip : ?axes:int list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval moveaxis : int -> int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval swapaxes : int -> int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval cat_tensors : axis:int -> ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval roll : ?axis:int -> int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval tile : int array -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval repeat : ?axis:int -> int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval check_dtypes_match : op:string -> ('a, 'b) B.t list -> unit

Sourceval concatenate : ?axis:int -> ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval stack : ?axis:IntSet.elt -> ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval ensure_ndim : int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval vstack : ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval hstack : ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval dstack : ('a, 'b) B.t list -> ('a, 'b) B.t

Sourceval broadcast_arrays : ('a, 'b) B.t list -> ('a, 'b) B.t list

Source

val eye : 
  B.context ->
  ?m:int ->
  ?k:int ->
  ('a, 'b) Dtype.t ->
  int ->
  ('a, 'b) B.t

Sourceval identity : B.context -> ('a, 'b) Dtype.t -> int -> ('a, 'b) B.t

Sourceval diag : ?k:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval arange : B.context -> ('a, 'b) Dtype.t -> int -> int -> int -> ('a, 'b) B.t

Source

val arange_f : 
  B.context ->
  (float, 'a) Dtype.t ->
  float ->
  float ->
  float ->
  (float, 'a) B.t

Source

val linspace : 
  B.context ->
  ('a, 'b) Dtype.t ->
  ?endpoint:bool ->
  float ->
  float ->
  int ->
  ('a, 'b) B.t

Source

val logspace : 
  B.context ->
  (float, 'a) Dtype.t ->
  ?endpoint:bool ->
  ?base:float ->
  float ->
  float ->
  int ->
  (float, 'a) B.t

Source

val geomspace : 
  B.context ->
  (float, 'a) Dtype.t ->
  ?endpoint:bool ->
  float ->
  float ->
  int ->
  (float, 'a) B.t

Source

val meshgrid : 
  ?indexing:[< `ij | `xy xy ] ->
  ('a, 'b) B.t ->
  ('c, 'd) B.t ->
  ('a, 'b) B.t * ('c, 'd) B.t

Source

val triangular_mask : 
  op:string ->
  cmp:
    ((int32, int32_elt) B.t ->
      (int32, int32_elt) B.t ->
      (bool, Dtype.bool_elt) B.t) ->
  ?k:int ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval tril : ?k:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval triu : ?k:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val apply_index_mode : 
  mode:[< `clip | `raise | `wrap ] ->
  n:int ->
  B.context ->
  (int32, int32_elt) B.t ->
  (int32, int32_elt) B.t

Source

val take : 
  ?axis:int ->
  ?mode:[< `clip | `raise | `wrap raise ] ->
  (int32, int32_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val take_along_axis : 
  axis:int ->
  (int32, Dtype.int32_elt) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval normalize_index : int -> int -> int

Sourceval normalize_and_check_index : op:string -> int -> int -> int

Sourcetype dim_op =

| View of {
1. start : int;
2. stop : int;
3. step : int;
4. dim_len : int;
}
| Squeeze of {
1. idx : int;
}
| Gather of int array
| New_axis

Sourceval normalize_slice_spec : int -> index -> dim_op

Sourceval slice_internal : index list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval set_slice_internal : index list -> ('a, 'b) B.t -> ('a, 'b) B.t -> unit

Sourceval get : int list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval set : int list -> ('a, 'b) B.t -> ('a, 'b) B.t -> unit

Sourceval unsafe_get : int list -> ('a, 'b) B.t -> 'a

Sourceval unsafe_set : int list -> 'a -> ('a, 'b) B.t -> unit

Sourceval slice : index list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval set_slice : index list -> ('a, 'b) B.t -> ('a, 'b) B.t -> unit

Sourceval item : int list -> ('a, 'b) B.t -> 'a

Sourceval set_item : int list -> 'a -> ('a, 'b) B.t -> unit

Source

val put : 
  ?axis:int ->
  indices:(int32, int32_elt) B.t ->
  values:('a, 'b) B.t ->
  ?mode:[< `clip | `raise | `wrap raise ] ->
  ('a, 'b) B.t ->
  unit

Source

val index_put : 
  indices:(int32, int32_elt) B.t array ->
  values:('a, 'b) B.t ->
  ?mode:[< `clip | `raise | `wrap raise ] ->
  ('a, 'b) B.t ->
  unit

Source

val put_along_axis : 
  axis:int ->
  indices:(int32, Dtype.int32_elt) B.t ->
  values:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  unit

Sourceval nonzero_indices_only : (bool, bool_elt) t -> (int32, int32_elt) B.t array

Source

val compress : 
  ?axis:int ->
  condition:(bool, bool_elt) t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval extract : condition:(bool, bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval nonzero : ('a, 'b) t -> (int32, int32_elt) B.t array

Sourceval argwhere : ('a, 'b) t -> (int32, int32_elt) B.t

Source

val array_split : 
  axis:int ->
  [< `Count of int | `Indices of int list ] ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t list

Sourceval split : axis:int -> int -> ('a, 'b) B.t -> ('a, 'b) B.t list

Source

val sort : 
  ?descending:bool ->
  ?axis:int ->
  ('a, 'b) t ->
  ('a, 'b) t * (int32, Dtype.int32_elt) B.t

Source

val argsort : 
  ?descending:bool ->
  ?axis:int ->
  ('a, 'b) t ->
  (int32, Dtype.int32_elt) B.t

Source

val argmax : 
  ?axis:int ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  (int32, Dtype.int32_elt) B.t

Source

val argmin : 
  ?axis:int ->
  ?keepdims:bool ->
  ('a, 'b) t ->
  (int32, Dtype.int32_elt) t

Source

val validate_random_float_params : 
  string ->
  ('a, 'b) Dtype.t ->
  Shape.t ->
  unit

Sourceval rand : B.context -> ('a, 'b) Dtype.t -> Shape.t -> ('a, 'b) B.t

Sourceval randn : B.context -> ('a, 'b) Dtype.t -> Shape.t -> ('a, 'b) B.t

Source

val randint : 
  B.context ->
  ('a, 'b) Dtype.t ->
  ?high:int ->
  Shape.t ->
  int ->
  ('a, 'b) t

Sourceval bernoulli : B.context -> p:float -> Shape.t -> (bool, Dtype.bool_elt) B.t

Sourceval permutation : B.context -> int -> (int32, Dtype.int32_elt) B.t

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

Source

val categorical : 
  B.context ->
  ?axis:int ->
  ?shape:int array ->
  ('a, 'b) t ->
  (int32, Dtype.int32_elt) t

Source

val truncated_normal : 
  B.context ->
  ('a, 'b) Dtype.t ->
  lower:float ->
  upper:float ->
  Shape.t ->
  ('a, 'b) B.t

Sourceval matmul_output_shape : ('a, 'b) B.t -> ('c, 'd) B.t -> int array

Source

val matmul_with_alloc : 
  ?out:('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval dot : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval matmul : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val diagonal : 
  ?offset:int ->
  ?axis1:int ->
  ?axis2:int ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval matrix_transpose : ('a, 'b) B.t -> ('a, 'b) B.t

Source

val extract_complex_part : 
  op:string ->
  field:(Complex.t -> float) ->
  ('a, 'b) t ->
  ('c, 'd) t

Sourceval complex : real:('a, 'b) t -> imag:('a, 'b) t -> 'c

Sourceval real : ('a, 'b) t -> ('c, 'd) t

Sourceval imag : ('a, 'b) t -> ('c, 'd) t

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

Sourceval vdot : ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) B.t

Sourceval vecdot : ?axis:int -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval inner : ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval outer : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val tensordot : 
  ?axes:(IntSet.elt list * IntSet.elt list) ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourcemodule Einsum : sig ... end

Sourceval einsum : string -> ('a, 'b) B.t array -> ('a, 'b) B.t

Sourceval kron : ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval multi_dot : ('a, 'b) B.t array -> ('a, 'b) B.t

Source

val cross : 
  ?out:('a, 'b) B.t ->
  ?axis:IntSet.elt ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval check_square : op:string -> ('a, 'b) B.t -> unit

Sourceval check_float_or_complex : op:string -> ('a, 'b) t -> unit

Sourceval check_real : op:string -> ('a, 'b) t -> unit

Sourceval cholesky : ?upper:bool -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val qr : 
  ?mode:[< `Complete | `Reduced ] ->
  ('a, 'b) t ->
  ('a, 'b) B.t * ('a, 'b) B.t

Source

val svd : 
  ?full_matrices:bool ->
  ('a, 'b) t ->
  ('a, 'b) B.t * (float, Dtype.float64_elt) B.t * ('a, 'b) B.t

Sourceval svdvals : ('a, 'b) t -> (float, Dtype.float64_elt) B.t

Source

val eig : 
  ('a, 'b) B.t ->
  (Complex.t, Dtype.complex64_elt) B.t * (Complex.t, Dtype.complex64_elt) B.t

Source

val eigh : 
  ?uplo:'a ->
  ('b, 'c) B.t ->
  (float, Dtype.float64_elt) B.t * ('b, 'c) B.t

Sourceval eigvals : ('a, 'b) B.t -> (Complex.t, Dtype.complex64_elt) B.t

Sourceval eigvalsh : ?uplo:'a -> ('b, 'c) B.t -> (float, Dtype.float64_elt) B.t

Source

val norm : 
  ?ord:
    [> `Fro
    | `Inf
    | `NegInf
    | `NegOne
    | `NegTwo
    | `Nuc
    | `One
    | `P of float
    | `Two ] ->
  ?axes:int list ->
  ?keepdims:bool ->
  ('a, 'b) t ->
  ('a, 'b) B.t

Source

val slogdet : 
  ('a, 'b) B.t ->
  (float, Dtype.float32_elt) B.t * (float, Dtype.float32_elt) t

Sourceval det : ('a, 'b) B.t -> ('a, 'b) B.t

Source

val matrix_rank : 
  ?tol:float ->
  ?rtol:float ->
  ?hermitian:bool ->
  ('a, 'b) t ->
  int

Sourceval trace : ?out:('a, 'b) B.t -> ?offset:int -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval solve : ('a, 'b) B.t -> ('a, 'b) t -> ('a, 'b) B.t

Sourceval pinv : ?rtol:float -> ?hermitian:bool -> ('a, 'b) t -> ('a, 'b) B.t

Source

val lstsq : 
  ?rcond:float ->
  ('a, 'b) t ->
  ('a, 'b) t ->
  ('a, 'b) B.t * ('a, 'b) B.t * int * (float, Dtype.float64_elt) B.t

Sourceval inv : ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval matrix_power : ('a, 'b) B.t -> int -> ('a, 'b) B.t

Sourceval cond : ?p:[> `Inf | `One | `Two ] -> ('a, 'b) B.t -> ('a, 'b) t

Sourceval tensorsolve : ?axes:int list -> ('a, 'b) t -> ('a, 'b) t -> ('a, 'b) B.t

Sourceval tensorinv : ?ind:int -> ('a, 'b) t -> ('a, 'b) B.t

Sourcetype fft_norm = [

| `Backward
| `Forward
| `Ortho

]

Source

val pad_or_truncate_for_fft : 
  ('a, 'b) B.t ->
  int list ->
  int list option ->
  ('a, 'b) B.t

Source

val fft_norm_scale : 
  [< `Backward | `Forward | `Ortho ] ->
  int list ->
  ('a, 'b) B.t ->
  float

Source

val ifft_norm_scale : 
  [< `Backward | `Forward | `Ortho ] ->
  int list ->
  ('a, 'b) B.t ->
  float

Source

val apply_fft_scale : 
  ?out:(Complex.t, 'a) B.t ->
  float ->
  (Complex.t, 'a) t ->
  (Complex.t, 'a) t

Source

val fft : 
  ?out:(Complex.t, 'a) B.t ->
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) t ->
  (Complex.t, 'a) t

Source

val ifft : 
  ?out:(Complex.t, 'a) B.t ->
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) t ->
  (Complex.t, 'a) t

Source

val rfft : 
  ?out:(Complex.t, Dtype.complex64_elt) B.t ->
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (float, 'a) B.t ->
  (Complex.t, Dtype.complex64_elt) t

Source

val irfft : 
  ?out:(float, Dtype.float64_elt) B.t ->
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (float, Dtype.float64_elt) B.t

Sourceval check_fft2 : op:string -> ('a, 'b) B.t -> int list option -> int list

Source

val fft2 : 
  ?out:(Complex.t, 'a) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (Complex.t, 'a) t

Source

val ifft2 : 
  ?out:(Complex.t, 'a) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (Complex.t, 'a) t

Source

val fftn : 
  ?out:(Complex.t, 'a) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (Complex.t, 'a) t

Source

val ifftn : 
  ?out:(Complex.t, 'a) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (Complex.t, 'a) t

Source

val rfft2 : 
  ?out:(Complex.t, Dtype.complex64_elt) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (float, 'a) B.t ->
  (Complex.t, Dtype.complex64_elt) t

Source

val irfft2 : 
  ?out:(float, Dtype.float64_elt) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (float, Dtype.float64_elt) B.t

Source

val rfftn : 
  ?out:(Complex.t, Dtype.complex64_elt) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (float, 'a) B.t ->
  (Complex.t, Dtype.complex64_elt) t

Source

val irfftn : 
  ?out:(float, Dtype.float64_elt) B.t ->
  ?axes:int list ->
  ?s:int list ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (float, Dtype.float64_elt) B.t

Source

val hfft : 
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (Complex.t, 'a) B.t ->
  (float, Dtype.float64_elt) B.t

Source

val ihfft : 
  ?axis:int ->
  ?n:int ->
  ?norm:[< `Backward | `Forward | `Ortho Backward ] ->
  (float, 'a) B.t ->
  (Complex.t, Dtype.complex64_elt) t

Sourceval fftfreq : B.context -> ?d:float -> int -> (float, Dtype.float64_elt) B.t

Sourceval rfftfreq : B.context -> ?d:float -> int -> (float, Dtype.float64_elt) B.t

Sourceval fftshift : ?axes:int list -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval ifftshift : ?axes:int list -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val softmax : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?scale:float ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val log_softmax : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?scale:float ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val logsumexp : 
  ?out:('a, 'b) B.t ->
  ?axes:IntSet.elt list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val logmeanexp : 
  ?out:('a, 'b) B.t ->
  ?axes:IntSet.elt list ->
  ?keepdims:bool ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val standardize : 
  ?out:('a, 'b) B.t ->
  ?axes:int list ->
  ?mean:('a, 'b) B.t ->
  ?variance:('a, 'b) B.t ->
  ?epsilon:float ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval erf : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t

Source

val extract_patches : 
  kernel_size:int array ->
  stride:int array ->
  dilation:int array ->
  padding:(int * int) array ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val combine_patches : 
  output_size:int array ->
  kernel_size:int array ->
  stride:int array ->
  dilation:int array ->
  padding:(int * int) array ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val correlate_padding : 
  mode:[< `Full | `Same | `Valid ] ->
  'a ->
  int array ->
  (int * int) array

Source

val correlate : 
  ?padding:[< `Full | `Same | `Valid Valid ] ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val convolve : 
  ?padding:[< `Full | `Same | `Valid Valid ] ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val sliding_filter : 
  reduce_fn:(('a, 'b) B.t -> axes:int list -> keepdims:bool -> ('c, 'd) B.t) ->
  kernel_size:int array ->
  ?stride:int array ->
  ('a, 'b) B.t ->
  ('c, 'd) B.t

Source

val maximum_filter : 
  kernel_size:int array ->
  ?stride:int array ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val minimum_filter : 
  kernel_size:int array ->
  ?stride:int array ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Source

val uniform_filter : 
  kernel_size:int array ->
  ?stride:int array ->
  ('a, 'b) B.t ->
  ('a, 'b) B.t

Sourceval one_hot : num_classes:int -> ('a, 'b) B.t -> (int, Dtype.uint8_elt) t

Sourceval pp_data : Format.formatter -> ('a, 'b) t -> unit

Sourceval format_to_string : (Format.formatter -> 'a -> 'b) -> 'a -> string

Sourceval print_with_formatter : (Format.formatter -> 'a -> 'b) -> 'a -> unit

Sourceval data_to_string : ('a, 'b) t -> string

Sourceval print_data : ('a, 'b) t -> unit

Sourceval pp_dtype : Format.formatter -> ('a, 'b) Dtype.t -> unit

Sourceval dtype_to_string : ('a, 'b) Dtype.t -> string

Sourceval shape_to_string : int array -> string

Sourceval pp_shape : Format.formatter -> int array -> unit

Sourceval pp : Format.formatter -> ('a, 'b) B.t -> unit

Sourceval print : ('a, 'b) B.t -> unit

Sourceval to_string : ('a, 'b) B.t -> string

Sourceval map_item : ('a -> 'a) -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval iter_item : ('a -> 'b) -> ('a, 'c) B.t -> unit

Sourceval fold_item : ('a -> 'b -> 'a) -> 'a -> ('b, 'c) B.t -> 'a

Sourceval map : (('a, 'b) B.t -> ('a, 'b) B.t) -> ('a, 'b) B.t -> ('a, 'b) B.t

Sourceval iter : (('a, 'b) B.t -> 'c) -> ('a, 'b) B.t -> unit

Sourceval fold : ('a -> ('b, 'c) B.t -> 'a) -> 'a -> ('b, 'c) B.t -> 'a

Sourcemodule Infix : sig ... end

package nx

Module Nx_core.Make_frontendSource

Parameters

Signature

Module `Nx_core.Make_frontend`Source