libtorch-ffi-1.11.0.0: test out alternative options for ffi interface to libtorch 1.x
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Internal.Managed.Autograd

Documentation

grad :: ForeignPtr Tensor -> ForeignPtr TensorList -> IO (ForeignPtr TensorList) Source #

makeIndependent :: ForeignPtr Tensor -> CBool -> IO (ForeignPtr Tensor) Source #

dropVariable :: ForeignPtr Tensor -> IO (ForeignPtr Tensor) Source #