Struct ActivationForward 

pub struct ActivationForward<'a, A: CudnnDataType, X: CudnnDataType, Y: CudnnDataType> {
    pub act: &'a ActivationDescriptor<A>,
    pub x: &'a TensorDescriptor<X>,
    pub y: &'a TensorDescriptor<Y>,
}

The activation forward operation. Pass in references to descriptors directly, and then call [ConvForward::launch()] .

Fields

act: &'a ActivationDescriptor<A>

Activation function.

x: &'a TensorDescriptor<X>

y: &'a TensorDescriptor<Y>

Implementations

impl<A, X, Y> ActivationForward<'_, A, X, Y>

where A: CudnnDataType, X: CudnnDataType, Y: CudnnDataType,

pub unsafe fn launch<Src, Dst>( &self, (alpha, beta): (Y, Y), x: &Src, y: &mut Dst, ) -> Result<(), CudnnError>

where Src: DevicePtr<A>, Dst: DevicePtrMut<A>,

Launches the operation.

• src is the input tensor
• y is the output

Safety

The arguments must match the data type/layout specified in the descriptors in `self.