Boilerplate code for the sum operator.

This commit is contained in:
laurent
2023-06-25 09:35:17 +01:00
parent 7ccf27dda2
commit 3852a85af3
7 changed files with 61 additions and 1 deletions

View File

@ -171,6 +171,15 @@ impl CpuStorage {
}
}
pub(crate) fn sum(
&self,
_shape: &Shape,
_stride: &[usize],
_sum_dims: &[usize],
) -> Result<Self> {
todo!()
}
pub(crate) fn divide_by_sum_over_dim(&mut self, shape: &Shape, dim: usize) -> Result<()> {
// [self] stores data in a contiguous way.
let dims = shape.dims();