hasktorch-0.2.0.0: Functional differentiable programming in Haskell
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Typed.Serialize

Synopsis

Documentation

save Source #

Arguments

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

list of input tensors

-> FilePath

file

-> IO () 

save list of tensors to file

load Source #

Arguments

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

file

-> IO (HList tensors) 

load list of tensors from file