kaun
Ptree.Dict
kaun.datasets
kaun.huggingface
kaun.models
type fields = (string * t) list
val fields_exn : ?ctx:string -> t -> fields
Extract dict fields or fail with a helpful ctx-prefixed message.
ctx
val find : string -> fields -> t option
val find_exn : ?ctx:string -> string -> fields -> t
val set : string -> t -> fields -> fields
val update : (t -> t) -> string -> fields -> fields
val mem : string -> fields -> bool
val get_tensor : fields -> name:string -> ('a, 'l) Nx_core.Dtype.t -> ('a, 'l) Rune.t option
val get_tensor_exn : fields -> name:string -> ('a, 'l) Nx_core.Dtype.t -> ('a, 'l) Rune.t