public INDArray eye(INDArray rows, INDArray cols) {
    NDValidation.validateInteger("eye", "rows", rows);
    NDValidation.validateInteger("eye", "cols", cols);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.shape.Eye(rows, cols))[0];
  }