def reduce(
self, op_name: str, X: BlockArray, axis=None, keepdims=False, dtype=None
):
res = X.reduce_axis(op_name, axis, keepdims=keepdims)
if dtype is not None:
res = res.astype(dtype)
return res
def sum(self, X: BlockArray, axis=0, keepdims=False):
return X.reduce_axis("sum", axis, keepdims=keepdims)
