Torch.Typed.NN.Dropout

Documentation

data DropoutSpec where

Constructors

DropoutSpec

Fields

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

Instances

Show DropoutSpec

Defined in Torch.Typed.NN.Dropout

Methods

showsPrec :: Int -> DropoutSpec -> ShowS

show :: DropoutSpec -> String

showList :: [DropoutSpec] -> ShowS

Eq DropoutSpec

Defined in Torch.Typed.NN.Dropout

Methods

(==) :: DropoutSpec -> DropoutSpec -> Bool

(/=) :: DropoutSpec -> DropoutSpec -> Bool

Randomizable DropoutSpec Dropout

Defined in Torch.Typed.NN.Dropout

Methods

sample :: DropoutSpec -> IO Dropout

data Dropout where

Constructors

Dropout

Fields

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

Instances

Generic Dropout

Defined in Torch.Typed.NN.Dropout

Associated Types

type Rep Dropout

Defined in Torch.Typed.NN.Dropout

type Rep Dropout = D1 ('MetaData "Dropout" "Torch.Typed.NN.Dropout" "hasktorch-0.2.1.2-inplace" '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

Defined in Torch.Typed.NN.Dropout

Methods

showsPrec :: Int -> Dropout -> ShowS

show :: Dropout -> String

showList :: [Dropout] -> ShowS

Parameterized Dropout

Defined in Torch.Typed.NN.Dropout

Associated Types

type Parameters Dropout

Defined in Torch.Typed.NN.Dropout

type Parameters Dropout = GParameters (Rep Dropout)

Methods

flattenParameters :: Dropout -> HList (Parameters Dropout)

replaceParameters :: Dropout -> HList (Parameters Dropout) -> Dropout

Randomizable DropoutSpec Dropout

Defined in Torch.Typed.NN.Dropout

Methods

sample :: DropoutSpec -> IO Dropout

HasForward Dropout (Tensor device dtype shape) (Tensor device dtype shape)

Defined in Torch.Typed.NN.Dropout

Methods

forward :: Dropout -> Tensor device dtype shape -> Tensor device dtype shape

forwardStoch :: Dropout -> Tensor device dtype shape -> IO (Tensor device dtype shape)

type Rep Dropout

Defined in Torch.Typed.NN.Dropout

type Rep Dropout = D1 ('MetaData "Dropout" "Torch.Typed.NN.Dropout" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "Dropout" 'PrefixI 'True) (S1 ('MetaSel ('Just "dropoutProb") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Double)))

type Parameters Dropout

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)