Struct rcudnn::utils::DropoutConfig

pub struct DropoutConfig { /* fields omitted */ }

Provides a convenient interface to access cuDNN’s Dropout Descriptor.

You wouldn’t use this struct yourself, but rather obtain it through Cudnn.init_dropout().

Implementations

impl DropoutConfig

pub fn new(
    dropout_desc: DropoutDescriptor,
    reserve: CudaDeviceMemory
) -> DropoutConfig

Returns a new DropoutConfig.

pub fn dropout_desc(&self) -> &DropoutDescriptor

Returns dropout_desc.

pub fn take_mem(self) -> CudaDeviceMemory

Take the Reserve Memory of the DropoutDescriptor

pub fn reserved_space(&self) -> &CudaDeviceMemory

Returns the reserved space.

Trait Implementations

impl Debug for DropoutConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.