def infer_shape(self, nodes, shapes):
return [(tensor.minimum(*shapes[0]), )]
微信
公众号
扫码关注公众号
微信
公众号