我已经为我的神经网络编写了一个自定义损失函数，但是它无法计算任何梯度。我认为这是因为我需要最大值的索引，因此正在使用argmax来获取此索引。
由于argmax不可区分，所以我解决了这个问题，但是我不知道这是怎么可能的。
有人可以帮忙吗？
python tensorflow
python ×1
tensorflow ×1