| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
Torch.Autograd
Synopsis
- newtype IndependentTensor = IndependentTensor {}
- data GradOptions = GradOptions {
- keepGraph :: Bool
- createGraph :: Bool
- accumulateGrad :: Bool
- grad :: Tensor -> [IndependentTensor] -> [Tensor]
- gradWithOptions :: GradOptions -> Tensor -> [IndependentTensor] -> [Tensor]
- requiresGrad :: Tensor -> Bool
- setRequiresGrad :: Bool -> Tensor -> Tensor
- makeIndependent :: Tensor -> IO IndependentTensor
- makeIndependentWithRequiresGrad :: Tensor -> Bool -> IO IndependentTensor
Documentation
newtype IndependentTensor Source #
Note: to create an IndependentTensor use makeIndependent;
| otherwise, Torch will complain the parameter does not require a gradient.
Constructors
| IndependentTensor | |
Fields | |
Instances
data GradOptions Source #
Constructors
| GradOptions | |
Fields
| |
Instances
| Show GradOptions Source # | |
Defined in Torch.Autograd Methods showsPrec :: Int -> GradOptions -> ShowS # show :: GradOptions -> String # showList :: [GradOptions] -> ShowS # | |
| Default GradOptions Source # | |
Defined in Torch.Autograd Methods def :: GradOptions # | |
gradWithOptions :: GradOptions -> Tensor -> [IndependentTensor] -> [Tensor] Source #
requiresGrad :: Tensor -> Bool Source #