Python tf.linalg.svd能否在Keras中的自定义损耗函数中使用？

我真的希望损失函数中有一些与输出的奇异值相关的项。能否使用

tf.linalg.svd

？我已经建立了我想要的损失函数，但它立即脱离轨道，使权重都

NaN

。我怀疑梯度计算不正确。损失函数在评估过程中起作用。

是的，

tf.linalg.svd

可用于损失函数，原始操作

svd

具有梯度，如图所示：

Op名称 有梯度 Svd ✔️

---

我拉了梯度，在我的情况下得到了所有的南。但我仍然是2.0.0版本的Tensorflow。这是什么版本的？编辑：我看到2.0.0版本的tf.raw_ops表明svd有梯度。我也可以复制你的结果。在我的损失函数的上下文中，关于svd的某些内容一定是问题所在。