Safe (p.nx.1.0.0~alpha2.doc.nx.io.Nx

Sourcetype error =

| Io_error of string
| Format_error of string
| Unsupported_dtype
| Unsupported_shape
| Missing_entry of string
| Other of string
(*
Custom error variants for I/O operations.
*)

Conversions from Packed Arrays

Sourceval as_float16 : packed_nx -> (Nx.float16_t, error) result

Safe alias for as_float16

Sourceval as_bfloat16 : packed_nx -> (Nx.bfloat16_t, error) result

Safe alias for as_bfloat16

Sourceval as_float32 : packed_nx -> (Nx.float32_t, error) result

Safe alias for as_float32

Sourceval as_float64 : packed_nx -> (Nx.float64_t, error) result

Safe alias for as_float64

Sourceval as_int8 : packed_nx -> (Nx.int8_t, error) result

Safe alias for as_int8

Sourceval as_int16 : packed_nx -> (Nx.int16_t, error) result

Safe alias for as_int16

Sourceval as_int32 : packed_nx -> (Nx.int32_t, error) result

Safe alias for as_int32

Sourceval as_int64 : packed_nx -> (Nx.int64_t, error) result

Safe alias for as_int64

Sourceval as_uint8 : packed_nx -> (Nx.uint8_t, error) result

Safe alias for as_uint8

Sourceval as_uint16 : packed_nx -> (Nx.uint16_t, error) result

Safe alias for as_uint16

Sourceval as_bool : packed_nx -> (Nx.bool_t, error) result

Safe alias for as_bool

Sourceval as_complex32 : packed_nx -> (Nx.complex32_t, error) result

Safe alias for as_complex32

Sourceval as_complex64 : packed_nx -> (Nx.complex64_t, error) result

Safe alias for as_complex64

Image Loading and Saving

Source

val load_image : 
  ?grayscale:bool ->
  string ->
  ((int, Nx.uint8_elt) Nx.t, error) result

Safe alias for load_image

Source

val save_image : 
  ?overwrite:bool ->
  string ->
  (int, Nx.uint8_elt) Nx.t ->
  (unit, error) result

Safe alias for save_image

NumPy Format (.npy)

Sourceval load_npy : string -> (packed_nx, error) result

Safe alias for load_npy

Source

val save_npy : 
  ?overwrite:bool ->
  string ->
  ('a, 'b) Nx.t ->
  (unit, error) result

Safe alias for save_npy

NumPy Archive Format (.npz)

Sourceval load_npz : string -> (archive, error) result

Safe alias for load_npz

Sourceval load_npz_member : name:string -> string -> (packed_nx, error) result

Safe alias for load_npz_member

Source

val save_npz : 
  ?overwrite:bool ->
  string ->
  (string * packed_nx) list ->
  (unit, error) result

Safe alias for save_npz

Text Format

Source

val save_txt : 
  ?sep:string ->
  ?append:bool ->
  ?newline:string ->
  ?header:string ->
  ?footer:string ->
  ?comments:string ->
  out:string ->
  ('a, 'b) Nx.t ->
  (unit, error) result

Safe alias for save_txt

Source

val load_txt : 
  ?sep:string ->
  ?comments:string ->
  ?skiprows:int ->
  ?max_rows:int ->
  ('a, 'b) Nx.dtype ->
  string ->
  (('a, 'b) Nx.t, error) result

Safe alias for load_txt

SafeTensors Format

Sourceval load_safetensor : string -> (archive, error) result

Safe alias for load_safetensor

Source

val save_safetensor : 
  ?overwrite:bool ->
  string ->
  (string * packed_nx) list ->
  (unit, error) result

Safe alias for save_safetensor