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

Torch.Jit

Documentation

newtype ScriptCache Source #

Constructors

ScriptCache 

Fields

newScriptCache :: IO ScriptCache Source #

jitIO :: ScriptCache -> ([Tensor] -> IO [Tensor]) -> [Tensor] -> IO [Tensor] Source #

jit :: ScriptCache -> ([Tensor] -> [Tensor]) -> [Tensor] -> [Tensor] Source #