def METHOD_NAME(cls, op):
    out = op.outputs[0]
    chunk_op = op.copy().reset_key()
    chunk_params = out.params
    chunk_params["index"] = (0,) * out.ndim
    out_chunk = chunk_op.new_chunk(op.inputs[0].chunks, kws=[chunk_params])
    new_op = op.copy()
    params = out.params
    params["nsplits"] = tuple((s,) for s in out.shape)
    params["chunks"] = [out_chunk]
    return new_op.new_tensors(op.inputs, kws=[params])