docs: public SDVariable eye(String name, SDVariable rows, SDVariable cols, SDVariable batchDimension) {
          SDVariable eye = new Eye(sd, rows, cols, batchDimension).outputVariable();
          return updateVariableNameAndReference(eye, name);
      }