| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
Torch.Typed.Autograd
Documentation
class HasGrad a b | a -> b Source #
Minimal complete definition
grad, gradWithOptions, toDependent
Instances
| (HasGrad a b, HasGrad (HList as) (HList bs), Castable (HList (b ': bs)) [ATenTensor]) => HasGrad (HList (a ': as)) (HList (b ': bs)) Source # | |
Defined in Torch.Typed.Autograd Methods grad :: forall (dtype :: DType) (device :: (DeviceType, Nat)). Tensor device dtype ('[] :: [Nat]) -> HList (a ': as) -> HList (b ': bs) Source # gradWithOptions :: forall (dtype :: DType) (device :: (DeviceType, Nat)). GradOptions -> Tensor device dtype ('[] :: [Nat]) -> HList (a ': as) -> HList (b ': bs) toDependent :: HList (a ': as) -> HList (b ': bs) | |
| HasGrad (HList ('[] :: [Type])) (HList ('[] :: [Type])) Source # | |
Defined in Torch.Typed.Autograd Methods grad :: forall (dtype :: DType) (device :: (DeviceType, Nat)). Tensor device dtype ('[] :: [Nat]) -> HList ('[] :: [Type]) -> HList ('[] :: [Type]) Source # gradWithOptions :: forall (dtype :: DType) (device :: (DeviceType, Nat)). GradOptions -> Tensor device dtype ('[] :: [Nat]) -> HList ('[] :: [Type]) -> HList ('[] :: [Type]) toDependent :: HList ('[] :: [Type]) -> HList ('[] :: [Type]) | |
| HasGrad (Parameter device dtype shape) (Tensor device dtype shape) Source # | |
Defined in Torch.Typed.Autograd Methods grad :: forall (dtype0 :: DType) (device0 :: (DeviceType, Nat)). Tensor device0 dtype0 ('[] :: [Nat]) -> Parameter device dtype shape -> Tensor device dtype shape Source # gradWithOptions :: forall (dtype0 :: DType) (device0 :: (DeviceType, Nat)). GradOptions -> Tensor device0 dtype0 ('[] :: [Nat]) -> Parameter device dtype shape -> Tensor device dtype shape toDependent :: Parameter device dtype shape -> Tensor device dtype shape | |