hasktorch-0.2.1.2: Haskell bindings to libtorch, supporting both typed and untyped tensors.
Safe HaskellNone
LanguageHaskell2010

Torch.Typed.Serialize

Synopsis

Documentation

save Source #

Arguments

:: forall (tensors :: [Type]). Castable (HList tensors) [ATenTensor] 
=> HList tensors

list of input tensors

-> FilePath

file

-> IO () 

save list of tensors to file

load Source #

Arguments

:: forall (tensors :: [Type]). Castable (HList tensors) [ATenTensor] 
=> FilePath

file

-> IO (HList tensors) 

load list of tensors from file