Forward, then calculate the loss, and update the weights. Finally return the average loss of this batch.
(&mut self, feature: &NdArray, label: &Vec<f32>, lr: f32, gradient_clip_by_norm: Option<NormType>)
source not stored for this graph (policy: none)