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