Dropout

TODO create cuRAND wrappers

struct Dropout (

T

size_t dim

) {

import numir.random : generate;;

import mir.ndslice : as, slice;;

import mir.random.variable : BernoulliVariable;;

Variable!(T, dim, HostStorage) hostMask;

this(double ratio);

auto forward(Variable!(T, dim, HostStorage) x);

auto backward(Variable!(T, dim, HostStorage) gy);

import grain.cudnn : CudnnDropout;;

version(grain_cuda)

CudnnDropout impl;

version(grain_cuda)

auto forward(Variable!(T, dim, DeviceStorage) x);

version(grain_cuda)

auto backward(Variable!(T, dim, DeviceStorage) gy);

mixin FunctionCommon ;

}

Constructors

this this(double ratio): Undocumented in source.

Members

Functions

backward auto backward(Variable!(T, dim, DeviceStorage) gy): Undocumented in source. Be warned that the author may not have intended to support it.
backward auto backward(Variable!(T, dim, HostStorage) gy): Undocumented in source. Be warned that the author may not have intended to support it.
forward auto forward(Variable!(T, dim, HostStorage) x): Undocumented in source. Be warned that the author may not have intended to support it.
forward auto forward(Variable!(T, dim, DeviceStorage) x): Undocumented in source. Be warned that the author may not have intended to support it.

Mixins

__anonymous mixin FunctionCommon: Undocumented in source.

Variables

hostMask Variable!(T, dim, HostStorage) hostMask;: Undocumented in source.
impl CudnnDropout impl;: Undocumented in source.
ratio float ratio;: Undocumented in source.

Mixed In Members

From `mixin FunctionCommon`

this(this) this(this): Undocumented in source.
DeviceRets alias DeviceRets = Tuple!(Parameters!backward): Undocumented in source.
DeviceArgs alias DeviceArgs = Tuple!(Parameters!forward): Undocumented in source.
__anonymous mixin TypeChecker!(forward, backward): Undocumented in source.
_mixin_dargs DeviceArgs _mixin_dargs;: Undocumented in source.
HostRets alias HostRets = Tuple!(Parameters!backward): Undocumented in source.
HostArgs alias HostArgs = Tuple!(Parameters!forward): Undocumented in source.
__anonymous mixin TypeChecker!(forward, backward): Undocumented in source.
_mixin_hargs HostArgs _mixin_hargs;: Undocumented in source.
applyForward auto applyForward(Args args): store grain.autograd.BackProp object in returned variables from forward function
applyBackward void applyBackward(UntypedVariable[] ugradOutputs): type-erased version of backward function used in grain.autograd.BackProp object

Meta

Source

See Implementation

grain functions random structs

Dropout