Beispiel #1
0
def reduce_min(x, axis = None):
  return reduce(prims.minimum, x, axis = axis)
Beispiel #2
0
def reduce_max(x, axis = None):
  return reduce(prims.maximum, x, axis = axis)
Beispiel #3
0
def builtin_sum(x, axis = None):
  return reduce(prims.add, x, init = 0, axis = axis)
Beispiel #4
0
def builtin_all(x, axis = None):
  return reduce(builtin_and, x, axis = axis, init = True)
Beispiel #5
0
def builtin_any(x, axis=None):
  return reduce(builtin_or, x, axis = axis, init = False)
Beispiel #6
0
def prod(x, axis=None):
    return reduce(prims.multiply, x, init=true, axis=axis)
Beispiel #7
0
def prod(x, axis=None):
  return reduce(prims.multiply, x, init=true, axis = axis)