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

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