Safe Haskell	Safe-Inferred
Language	Haskell2010

Torch.GraduallyTyped.NN.Loss

Documentation

data MSELoss Source #

Constructors

MSELoss

Instances

Instances details

Generic MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Associated Types type Rep MSELoss :: Type -> Type Source # Methods from :: MSELoss -> Rep MSELoss x Source # to :: Rep MSELoss x -> MSELoss Source #
Show MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods showsPrec :: Int -> MSELoss -> ShowS Source # show :: MSELoss -> String Source # showList :: [MSELoss] -> ShowS Source #
Eq MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods (==) :: MSELoss -> MSELoss -> Bool Source # (/=) :: MSELoss -> MSELoss -> Bool Source #
Ord MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods compare :: MSELoss -> MSELoss -> Ordering Source # (<) :: MSELoss -> MSELoss -> Bool Source # (<=) :: MSELoss -> MSELoss -> Bool Source # (>) :: MSELoss -> MSELoss -> Bool Source # (>=) :: MSELoss -> MSELoss -> Bool Source # max :: MSELoss -> MSELoss -> MSELoss Source # min :: MSELoss -> MSELoss -> MSELoss Source #
HasStateDict MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods fromStateDict :: (MonadIO m, MonadThrow m, MonadState StateDict m) => ModelSpec MSELoss -> StateDictKey -> m MSELoss Source # toStateDict :: (MonadThrow m, MonadState StateDict m) => StateDictKey -> MSELoss -> m () Source #
HasInitialize MSELoss generatorDevice MSELoss generatorDevice Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods initialize :: MonadThrow m => ModelSpec MSELoss -> Generator generatorDevice -> m (MSELoss, Generator generatorDevice) Source #
(Catch (predShape <+> targetShape), output ~ Tensor (predGradient <\|> targetGradient) (predLayout <+> targetLayout) (predDevice <+> targetDevice) (predDataType <+> targetDataType) ('Shape ('[] :: [Dim (Name Symbol) (Size Nat)]))) => HasForward MSELoss (Tensor predGradient predLayout predDevice predDataType predShape, Tensor targetGradient targetLayout targetDevice targetDataType targetShape) generatorDevice output generatorDevice Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss Methods forward :: MonadThrow m => MSELoss -> (Tensor predGradient predLayout predDevice predDataType predShape, Tensor targetGradient targetLayout targetDevice targetDataType targetShape) -> Generator generatorDevice -> m (output, Generator generatorDevice) Source #
type Rep MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss type Rep MSELoss = D1 ('MetaData "MSELoss" "Torch.GraduallyTyped.NN.Loss" "hasktorch-gradually-typed-0.2.0.0-1KV1aIPzzbp6JpSr37tC1K" 'False) (C1 ('MetaCons "MSELoss" 'PrefixI 'False) (U1 :: Type -> Type))
type ModelSpec MSELoss Source #
Instance details Defined in Torch.GraduallyTyped.NN.Loss type ModelSpec MSELoss = MSELoss