mxnet infer_shape_entry TypeError：需要一个整数

Question

我在测试自定义运算符时遇到了 infer_shape 错误。

错误信息：

Error in <my_operator_name>.infer_shape: Traceback (most recent call last):
  File "~/anaconda2/envs/mx110-py27/lib/python2.7/site-packages/mxnet/operator.py", line 658, in infer_shape_entry
    array('I', rshape[i])),
TypeError: an integer is required

以前有人遇到过同样的问题吗？谢谢！

Answer 1

我发现这是因为 mxnet.operator.CustomProp 的 __init__ 函数只接受字符串类型的参数。