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