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)