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