package nx

  1. Overview
  2. Docs

Module Nx_io.SafeSource

Sourcetype error =
  1. | Io_error of string
  2. | Format_error of string
  3. | Unsupported_dtype
  4. | Unsupported_shape
  5. | Missing_entry of string
  6. | 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_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

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_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

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

Safe alias for load_image

Sourceval 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

Sourceval 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

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

Safe alias for save_npz

SafeTensors Format

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

Safe alias for load_safetensor

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

Safe alias for save_safetensor