hasktorch-0.2.1.1: Haskell bindings to libtorch, supporting both typed and untyped tensors.
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 #