Op("diag") {
        javaPackage = "org.nd4j.linalg.api.ops.impl.shape"
        javaOpClass = "Diag"

        Input(DataType.NUMERIC, "input") {"Input tensor"}