Struct ScaledAddFunction 

pub struct ScaledAddFunction { /* private fields */ }

Example custom function with context: Scaled addition

Implementations

impl ScaledAddFunction

pub fn new(scale: f32) -> Self

Trait Implementations

impl CustomAutogradFunctionWithContext for ScaledAddFunction

fn forward( &self, ctx: &mut AutogradContext, inputs: &[Tensor], ) -> TorshResult<Vec<Tensor>>

Forward pass with context

fn backward( &self, ctx: &AutogradContext, grad_outputs: &[Tensor], ) -> TorshResult<Vec<Option<Tensor>>>

Backward pass with context

fn num_inputs(&self) -> usize

Number of inputs expected by this function

fn num_outputs(&self) -> usize

Number of outputs produced by this function

fn name(&self) -> &str

Get the name of this function (for debugging)