Exemplo n.º 1
0
 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]]
Exemplo n.º 2
0
 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]]