hasktorch-gradually-typed-0.2.0.0: experimental project for hasktorch
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.GraduallyTyped.NN.Dropout

Synopsis

Documentation

newtype Dropout where Source #

Given a random generator, randomly zeroes some of the elements of the input tensor with probability p using samples from a Bernoulli distribution. Each channel will be zeroed out independently on every forward call.

Constructors

Dropout 

Fields

Instances

Instances details
Generic Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

Associated Types

type Rep Dropout :: Type -> Type Source #

Show Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

Eq Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

Ord Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

HasStateDict Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

HasInitialize Dropout generatorDevice Dropout generatorDevice Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

Methods

initialize :: MonadThrow m => ModelSpec Dropout -> Generator generatorDevice -> m (Dropout, Generator generatorDevice) Source #

(input ~ Tensor gradient layout device dataType shape, output ~ Tensor gradient layout (device <+> generatorDevice) dataType shape, generatorOutputDevice ~ (device <+> generatorDevice), SGetDevice device, SGetGeneratorDevice generatorDevice) => HasForward Dropout input generatorDevice output generatorOutputDevice Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

Methods

forward :: MonadThrow m => Dropout -> input -> Generator generatorDevice -> m (output, Generator generatorOutputDevice) Source #

type Rep Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout

type Rep Dropout = D1 ('MetaData "Dropout" "Torch.GraduallyTyped.NN.Dropout" "hasktorch-gradually-typed-0.2.0.0-1KV1aIPzzbp6JpSr37tC1K" 'True) (C1 ('MetaCons "Dropout" 'PrefixI 'False) (S1 ('MetaSel ('Nothing :: Maybe Symbol) 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedLazy) (Rec0 Double)))
type ModelSpec Dropout Source # 
Instance details

Defined in Torch.GraduallyTyped.NN.Dropout