我的问题是关于自定义梯度计算。我有一个自定义图层，无法弄清楚如何正确编程其渐变。该梯度对应于类似于硬限制器（或具有非常陡峭过渡的S形）的传递函数：如果绝对值y=x小于阈值{{1}，则输出为|x| < th }或其他th。因此，如果sign(x)或g，则梯度g=0为|x| >= th。在搜索Web之后，我尝试了如下所述的_MySignGrad函数。代码可以运行，但是我遇到两个问题：1）我尝试“注册” grad计算，但是由于某种原因它没有被调用； 2）每当调用它时，由于代码可能是错误的，我确信我会遇到运行时错误。有提示吗？
g=1