我使用本教程和问题在 Tensorflow 中编写了一个自定义运算符。它与两个张量完美配合。但是当我将它插入CNN 模型时，它无法正常工作。
我不需要更新这个自定义操作的权重，因为这是一个固定的操作。因此，我尝试使用tf.RegisterGradient和ops.RegisterGradient来传递相同的梯度（[grad, grad]）、零和无（[None, None]）。我什至尝试过使用 ops.NotDifferentiable。但问题依然存在。
我的猜测是我无法正确使用渐变。因为，没有这部分，CNN 模型可以完美运行，但是当我添加它时，输出图像变为black或gray。
我怎样才能正确地做到这一点？