package torch

  1. Overview
  2. Docs
include module type of Torch_core.Wrapper.Serialize
val save : Torch_core.Wrapper.Tensor.t -> filename:string -> unit
val load : filename:string -> Torch_core.Wrapper.Tensor.t
val save_multi : named_tensors:(string * Torch_core.Wrapper.Tensor.t) list -> filename:string -> unit
val load_multi : names:string list -> filename:string -> Torch_core.Wrapper.Tensor.t list
val load_multi_ : named_tensors:(string * Torch_core.Wrapper.Tensor.t) list -> filename:string -> unit
val load_all : filename:string -> (string * Torch_core.Wrapper.Tensor.t) list
OCaml

Innovation. Community. Security.