Op("eye") {
        javaPackage = "org.nd4j.linalg.api.ops.impl.shape"
        Input(INT, "rows") { description = "Number of rows" }
        Input(INT, "cols") { description = "Number of columns" }
        Output(NUMERIC, "output"){ description = "Identity matrix" }