pub trait GradientFunction: Debug { // Required methods fn backward(&self); fn prev(&self) -> Vec<&Tensor>; }