Safe Haskell	None
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 # show :: DropoutSpec -> String # showList :: [DropoutSpec] -> ShowS #
Eq DropoutSpec Source #
Instance details Defined in Torch.Typed.NN.Dropout Methods (==) :: DropoutSpec -> DropoutSpec -> Bool # (/=) :: DropoutSpec -> DropoutSpec -> Bool #
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
Instance details Defined in Torch.Typed.NN.Dropout type Rep Dropout = D1 ('MetaData "Dropout" "Torch.Typed.NN.Dropout" "hasktorch-0.2.1.6-LDjqZXcNPYQAOVyrR3h8rq" 'False) (C1 ('MetaCons "Dropout" 'PrefixI 'True) (S1 ('MetaSel ('Just "dropoutProb") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Double)))

Methods

from :: Dropout -> Rep Dropout x #

to :: Rep Dropout x -> Dropout #

Show Dropout Source #

Instance details

Defined in Torch.Typed.NN.Dropout

Methods

showsPrec :: Int -> Dropout -> ShowS #

show :: Dropout -> String #

showList :: [Dropout] -> ShowS #

Parameterized Dropout Source #

Instance details

Defined in Torch.Typed.NN.Dropout

Associated Types

type Parameters Dropout
Instance details Defined in Torch.Typed.NN.Dropout type Parameters Dropout = GParameters (Rep Dropout)

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.1.6-LDjqZXcNPYQAOVyrR3h8rq" '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 :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). Dropout -> Bool -> Tensor device dtype shape -> IO (Tensor device dtype shape) Source #