val unsortedSegmentOpInput = Mixin("unsortedSegmentOp") {
        javaPackage = "org.nd4j.linalg.api.ops.impl.transforms.segment"
        Input(NUMERIC, "data") { description = "Data (variable) to perform unsorted segment max on" }
        Input(NUMERIC, "segmentIds") { description = "Variable for the segment IDs" }
        Input(INT, "numSegments") { description = "Number of segments" }