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