code: public SDVariable eye(String name, SDVariable rows, SDVariable cols, SDVariable batchDimension) {
          SDVariable eye = new Eye(sd, rows, cols, batchDimension).outputVariable();
          return updateVariableNameAndReference(eye, name);
      }
docs: As per {@link #eye(String, int, int, int...)} bit with the number of rows/columns specified as scalar SDVariables,
  and the batch dimension specified as a 1D SDVariable