def infer_shape(self, node, ins_shapes): x = node.inputs[0] m = basic.max(x) + 1 if self.minlength is not None: m = basic.maximum(m, self.minlength) return [[m]]