package torch

  1. Overview
  2. Docs
type t

A VarStore is used to store all the variables used by a given model. The model creates variables by calling Var_store.new_var for which it has to provide a name. Var_store.sub creates a sub-directory in the var store which is useful to group some variables together.

val create : ?frozen:bool -> ?device:Device.t -> name:string -> unit -> t
val sub : t -> string -> t
val (/) : t -> string -> t
val trainable_vars : t -> Tensor.t list
val all_vars : t -> (string * Tensor.t) list
val name : t -> string
val device : t -> Device.t
module Init : sig ... end
val new_var : ?trainable:bool -> t -> shape:int list -> init:Init.t -> name:string -> Tensor.t
val new_var_copy : ?trainable:bool -> t -> src:Tensor.t -> name:string -> Tensor.t
val freeze : t -> unit
val unfreeze : t -> unit
val copy : src:t -> dst:t -> unit
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events Blog Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct