Safe Haskell	Safe-Inferred
Language	Haskell2010

Torch.Typed.NN.Dropout

Documentation

data DropoutSpec where Source #

Constructors

DropoutSpec
Fields :: { dropoutProbSpec :: Double } -> DropoutSpec

Instances

Instances details

Show DropoutSpec Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods showsPrec :: Int -> DropoutSpec -> ShowS Source # show :: DropoutSpec -> String Source # showList :: [DropoutSpec] -> ShowS Source #
Eq DropoutSpec Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods (==) :: DropoutSpec -> DropoutSpec -> Bool Source # (/=) :: DropoutSpec -> DropoutSpec -> Bool Source #
Randomizable DropoutSpec Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods sample :: DropoutSpec -> IO Dropout Source #

data Dropout where Source #

Constructors

Dropout
Fields :: { dropoutProb :: Double } -> Dropout

Instances

Instances details

Generic Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout Associated Types type Rep Dropout :: Type -> Type Source # Methods from :: Dropout -> Rep Dropout x Source # to :: Rep Dropout x -> Dropout Source #
Show Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods showsPrec :: Int -> Dropout -> ShowS Source # show :: Dropout -> String Source # showList :: [Dropout] -> ShowS Source #
Parameterized Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout Associated Types type Parameters Dropout :: [Type] Source # Methods flattenParameters :: Dropout -> HList (Parameters Dropout) Source # replaceParameters :: Dropout -> HList (Parameters Dropout) -> Dropout Source #
Randomizable DropoutSpec Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods sample :: DropoutSpec -> IO Dropout Source #
HasForward Dropout (Tensor device dtype shape) (Tensor device dtype shape) Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods forward :: Dropout -> Tensor device dtype shape -> Tensor device dtype shape Source # forwardStoch :: Dropout -> Tensor device dtype shape -> IO (Tensor device dtype shape) Source #
type Rep Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout type Rep Dropout = D1 ('MetaData "Dropout" "Torch.Typed.NN.Dropout" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Dropout" 'PrefixI 'True) (S1 ('MetaSel ('Just "dropoutProb") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Double)))
type Parameters Dropout Source #
Instance details Defined in Torch.Typed.NN.Dropout type Parameters Dropout = GParameters (Rep Dropout)

dropoutForward :: forall shape dtype device. Dropout -> Bool -> Tensor device dtype shape -> IO (Tensor device dtype shape) Source #