在使用关键字传递参数时,在前向传递时调用了带有 ArrayBox 对象的原始函数

为了提供上下文，我正在处理一个项目，其中我需要使用一些不支持 autograd 的函数，这些函数不支持 autograd 的 ArrayBox 参数，但它们的导数很容易实现，因此我可以直接实现它们并扩展 autograd。 当这样做时，我遇到了间歇性问题（最糟糕的问题），这些函数有时仍然以 ArrayBox 参数调用，尽管它们被定义为原语，而且当发生这种情况时，就会引发错误。