package nx

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

Install 

dune-project
 Dependency

github.com Readme Changelog Edit opam file Versions (4)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

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

doc/nx.core/Nx_core/Make_frontend/index.html

Module Nx_core.Make_frontendSource

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 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 =
  1. | Float16 : (float, float16_elt) dtype
  2. | Float32 : (float, float32_elt) dtype
  3. | Float64 : (float, float64_elt) dtype
  4. | BFloat16 : (float, bfloat16_elt) dtype
  5. | Float8_e4m3 : (float, float8_e4m3_elt) dtype
  6. | Float8_e5m2 : (float, float8_e5m2_elt) dtype
  7. | Int4 : (int, int4_elt) dtype
  8. | UInt4 : (int, uint4_elt) dtype
  9. | Int8 : (int, int8_elt) dtype
  10. | UInt8 : (int, uint8_elt) dtype
  11. | Int16 : (int, int16_elt) dtype
  12. | UInt16 : (int, uint16_elt) dtype
  13. | Int32 : (int32, int32_elt) dtype
  14. | UInt32 : (int32, uint32_elt) dtype
  15. | Int64 : (int64, int64_elt) dtype
  16. | UInt64 : (int64, uint64_elt) dtype
  17. | Complex64 : (Complex.t, complex32_elt) dtype
  18. | Complex128 : (Complex.t, complex64_elt) dtype
  19. | 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 complex128 : (Complex.t, complex64_elt) dtype
Sourceval bool : (bool, bool_elt) dtype
Sourcetype index =
  1. | I of int
  2. | L of int list
  3. | R of int * int
  4. | Rs of int * int * int
  5. | A
  6. | M of (bool, bool_elt) t
  7. | 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
Sourceval 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
Sourceval create : B.context -> ('a, 'b) Dtype.t -> int array -> 'a array -> ('a, 'b) B.t
Sourceval 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
Sourceval 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
Sourceval to_bigarray : ('c, 'd) B.t -> ('a, 'b, Bigarray.c_layout) Bigarray.Genarray.t
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval bitwise_xor : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval bitwise_or : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval bitwise_and : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval logical_and : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval logical_or : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval cmpeq : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval cmpne : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval cmplt : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval cmple : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval cmpgt : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval cmpge : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval less : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval less_equal : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval greater : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval greater_equal : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval equal : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval not_equal : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval equal_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval not_equal_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval less_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval greater_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval less_equal_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval greater_equal_s : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> 'a -> (bool, Dtype.bool_elt) B.t
Sourceval 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
Sourceval isinf : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval isnan : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval isfinite : ?out:(bool, Dtype.bool_elt) B.t -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval lerp : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval lerp_scalar_weight : ?out:('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t -> 'a -> ('a, 'b) B.t
Sourceval 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
Sourceval clamp : ?out:('a, 'b) B.t -> ?min:'a -> ?max:'a -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval clip : ?out:('a, 'b) B.t -> ?min:'a -> ?max:'a -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval 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
Sourceval sum : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval max : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval min : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval prod : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval associative_scan : axis:int -> [ `Max | `Min | `Prod | `Sum ] -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval mean : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval var : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ?ddof:int -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval std : ?out:('a, 'b) B.t -> ?axes:int list -> ?keepdims:bool -> ?ddof:int -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval all : ?out:(bool, Dtype.bool_elt) B.t -> ?axes:int list -> ?keepdims:bool -> ('a, 'b) B.t -> (bool, Dtype.bool_elt) B.t
Sourceval 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
Sourceval 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
Sourceval arange_f : B.context -> (float, 'a) Dtype.t -> float -> float -> float -> (float, 'a) B.t
Sourceval linspace : B.context -> ('a, 'b) Dtype.t -> ?endpoint:bool -> float -> float -> int -> ('a, 'b) B.t
Sourceval logspace : B.context -> (float, 'a) Dtype.t -> ?endpoint:bool -> ?base:float -> float -> float -> int -> (float, 'a) B.t
Sourceval geomspace : B.context -> (float, 'a) Dtype.t -> ?endpoint:bool -> float -> float -> int -> (float, 'a) B.t
Sourceval meshgrid : ?indexing:[< `ij | `xy xy ] -> ('a, 'b) B.t -> ('c, 'd) B.t -> ('a, 'b) B.t * ('c, 'd) B.t
Sourceval 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
Sourceval apply_index_mode : mode:[< `clip | `raise | `wrap ] -> n:int -> B.context -> (int32, int32_elt) B.t -> (int32, int32_elt) B.t
Sourceval take : ?axis:int -> ?mode:[< `clip | `raise | `wrap raise ] -> (int32, int32_elt) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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 =
  1. | View of {
    1. start : int;
    2. stop : int;
    3. step : int;
    4. dim_len : int;
    }
  2. | Squeeze of {
    1. idx : int;
    }
  3. | Gather of int array
  4. | 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
Sourceval put : ?axis:int -> indices:(int32, int32_elt) B.t -> values:('a, 'b) B.t -> ?mode:[< `clip | `raise | `wrap raise ] -> ('a, 'b) B.t -> unit
Sourceval index_put : indices:(int32, int32_elt) B.t array -> values:('a, 'b) B.t -> ?mode:[< `clip | `raise | `wrap raise ] -> ('a, 'b) B.t -> unit
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval sort : ?descending:bool -> ?axis:int -> ('a, 'b) t -> ('a, 'b) t * (int32, Dtype.int32_elt) B.t
Sourceval argsort : ?descending:bool -> ?axis:int -> ('a, 'b) t -> (int32, Dtype.int32_elt) B.t
Sourceval argmax : ?axis:int -> ?keepdims:bool -> ('a, 'b) B.t -> (int32, Dtype.int32_elt) B.t
Sourceval argmin : ?axis:int -> ?keepdims:bool -> ('a, 'b) t -> (int32, Dtype.int32_elt) t
Sourceval 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
Sourceval 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
Sourceval categorical : B.context -> ?axis:int -> ?shape:int array -> ('a, 'b) t -> (int32, Dtype.int32_elt) t
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval qr : ?mode:[< `Complete | `Reduced ] -> ('a, 'b) t -> ('a, 'b) B.t * ('a, 'b) B.t
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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 = [
  1. | `Backward
  2. | `Forward
  3. | `Ortho
]
Sourceval pad_or_truncate_for_fft : ('a, 'b) B.t -> int list -> int list option -> ('a, 'b) B.t
Sourceval fft_norm_scale : [< `Backward | `Forward | `Ortho ] -> int list -> ('a, 'b) B.t -> float
Sourceval ifft_norm_scale : [< `Backward | `Forward | `Ortho ] -> int list -> ('a, 'b) B.t -> float
Sourceval apply_fft_scale : ?out:(Complex.t, 'a) B.t -> float -> (Complex.t, 'a) t -> (Complex.t, 'a) t
Sourceval fft : ?out:(Complex.t, 'a) B.t -> ?axis:int -> ?n:int -> ?norm:[< `Backward | `Forward | `Ortho Backward ] -> (Complex.t, 'a) t -> (Complex.t, 'a) t
Sourceval ifft : ?out:(Complex.t, 'a) B.t -> ?axis:int -> ?n:int -> ?norm:[< `Backward | `Forward | `Ortho Backward ] -> (Complex.t, 'a) t -> (Complex.t, 'a) t
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval 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
Sourceval hfft : ?axis:int -> ?n:int -> ?norm:[< `Backward | `Forward | `Ortho Backward ] -> (Complex.t, 'a) B.t -> (float, Dtype.float64_elt) B.t
Sourceval 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
Sourceval softmax : ?out:('a, 'b) B.t -> ?axes:int list -> ?scale:float -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval log_softmax : ?out:('a, 'b) B.t -> ?axes:int list -> ?scale:float -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval logsumexp : ?out:('a, 'b) B.t -> ?axes:IntSet.elt list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval logmeanexp : ?out:('a, 'b) B.t -> ?axes:IntSet.elt list -> ?keepdims:bool -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval extract_patches : kernel_size:int array -> stride:int array -> dilation:int array -> padding:(int * int) array -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval correlate_padding : mode:[< `Full | `Same | `Valid ] -> 'a -> int array -> (int * int) array
Sourceval correlate : ?padding:[< `Full | `Same | `Valid Valid ] -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval convolve : ?padding:[< `Full | `Same | `Valid Valid ] -> ('a, 'b) B.t -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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
Sourceval maximum_filter : kernel_size:int array -> ?stride:int array -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval minimum_filter : kernel_size:int array -> ?stride:int array -> ('a, 'b) B.t -> ('a, 'b) B.t
Sourceval 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