在编写自定义op时，如果输出形状是动态的，如何定义infer_shape函数的输出？

Question

我现在正在编写自定义操作，它的输出形状是动态的，如何在 infer_shape 函数中定义 output_shape 。

Answer 1

infer_shape的{​​{1}}函数有一个CustomOpProp参数。您可以使用它来动态计算output_shape。

如果您无法在没有输入数据的情况下推断输入形状，但可以推断出最大形状，那么您可以在此处注释：https://github.com/apache/incubator-mxnet/issues/9758