Struct nncombinator::cuda::kernel::activation::ActivationBackwardArgs

pub struct ActivationBackwardArgs<T>where
    T: DataTypeInfo,{
    pub loss: CudaPtr<T>,
    /* private fields */
}

Create an instance of an object representing the argument list during error back propagation of the activation function.

Fields

loss: CudaPtr<T>

loss value

Implementations

impl<T> ActivationBackwardArgs<T>where
    T: DataTypeInfo,

Create an instance of an object representing the list of arguments during error back propagation of the activation function.

pub fn new(
    u: CudaPtr<T>,
    loss: CudaPtr<T>,
    units_len: usize,
    batch_len: usize
) -> ActivationBackwardArgs<T>

Create a ActivationBackwardArgs instance

Arguments

• u - Input values from upper layers
• loss - loss value
• units_len - count of inputs and outputs of linear layer weights
• batch_len - batch count

Trait Implementations

impl<T> KernelArgs for ActivationBackwardArgs<T>where
    T: DataTypeInfo,

fn as_vec(&mut self) -> Vec<&mut dyn AsMutKernelPtr>

Returns a Vec<&mut dyn AsMutKernelPtr> of the type implementing AsMutKernelPtr, which is converted to a data type that can be passed to the cuda kernel in subsequent processing.