写自定义op时，如果输出shape是动态的，如何定义infer_shape函数的输出？
When writing a custom op, if the output shape is dynamic, how to define the output of infer_shape function?
我现在正在写一个自定义操作，它的输出形状是动态的，如何在[=18=中定义output_shape ]函数。
CustomOpProp 的 infer_shape 函数有一个 in_shape 参数。您可以使用它来动态计算 output_shape。
如果没有输入数据无法推断输入形状，但可以推断出最大形状，那么您可以按照此处的评论使用它：https://github.com/apache/incubator-mxnet/issues/9758
我现在正在写一个自定义操作，它的输出形状是动态的，如何在[=18=中定义output_shape ]函数。
CustomOpProp 的 infer_shape 函数有一个 in_shape 参数。您可以使用它来动态计算 output_shape。
如果没有输入数据无法推断输入形状，但可以推断出最大形状，那么您可以按照此处的评论使用它：https://github.com/apache/incubator-mxnet/issues/9758