Op("topK") {
        javaPackage = "org.nd4j.linalg.api.ops.impl.transforms.custom"
        Input(NUMERIC, "input") { description = "Input data" }
        Arg(NUMERIC, "k") { description = "The number of values to return" }
        Arg(BOOL, "sorted") { description = "Whether to return the values sorted or not" }