Skip to main content

sum_rows

Function sum_rows 

Source
pub fn sum_rows(x: &Tensor) -> Result<Tensor>
Available on crate feature train only.
Expand description

Column sums over all leading dims: [.., cols] -> [cols]. Bias gradients.