При написании пользовательской операции, если форма вывода является динамической, как определить вывод функции infer_shape?

Функция infer_shape функции CustomOpProp имеет параметр in_shape. Вы можете использовать его для динамического вычисления output_shape.

Если вы не можете вывести входную форму без входных данных, но можете вывести максимальную форму, вы можете использовать ее, как указано здесь: https://github.com/apache/incubator-mxnet/issues/9758