pub fn sum_rows(x: &Tensor) -> Result<Tensor>
Column sums over all leading dims: [.., cols] -> [cols]. Bias gradients.
[.., cols]
[cols]