def cumprod(x, axis = None):
  return scan(prims.multiply, x, axis = axis)
Exemple #2
0
def cumprod(x, axis=None):
    return scan(prims.multiply, x, axis=axis)
def cumsum(x, axis = None):
  return scan(prims.add, x, axis = axis)
Exemple #4
0
def cumsum(x, axis=None):
    return scan(prims.add, x, axis=axis)