code: 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);
      }
docs: Generate an identity matrix with the specified number of rows and columns, with optional leading dims<br>
  Example:<br>
  batchShape: [3,3]<br>
  numRows: 2<br>
  numCols: 4<br>
  returns a tensor of shape (3, 3, 2, 4) that consists of 3 * 3 batches of (2,4)-shaped identity matrices:<br>
  1 0 0 0<br>
  0 1 0 0<br>
  
  @param rows           Number of rows
  @param cols           Number of columns
  @param batchDimension Batch dimensions. May be null