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

作者: Moso31
发布时间: 2024-08-17 08:14:58 (1月前)
转自：

2 条回复

0#
回复此人
！啊啊 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 该 <code> infer_shape </code> 的功能 <code> CustomOpProp </code> 有一个 <code> in_shape </code> 参数。您可以使用它来动态计算output_shape。 </p> <P> 如果您无法在没有输入数据的情况下推断输入形状，但可以推断最大形状，那么您可以在此处使用它作为评论： <a href="https://github.com/apache/incubator-mxnet/issues/9758" rel="nofollow noreferrer"> https://github.com/apache/incubator-mxnet/issues/9758 </A> </p> </DIV>

编辑

登录后才能参与评论