Skip to main content

sum_rows

Function sum_rows 

Source
pub fn sum_rows(x: &Tensor) -> Result<Tensor>
Expand description

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