docs: public SDVariable eye(String name, int rows, int cols, DataType dataType, int... batchDimension) {
          SDVariable eye = new Eye(sd, rows, cols, dataType, batchDimension).outputVariables()[0];
          return updateVariableNameAndReference(eye, name);
      }