libtorch-ffi-2.0.1.10: Haskell bindings for PyTorch
Safe HaskellNone
LanguageHaskell2010

Torch.Internal.Unmanaged.Autograd

Documentation

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

gradWithOptions :: CBool -> CBool -> CBool -> Ptr Tensor -> Ptr TensorList -> IO (Ptr TensorList) Source #

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

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