我真的希望我的损失函数有一些与输出奇异值相关的项。可以使用tf.linalg.svd吗？我已经建立了我想要的损失函数，但它立即偏离轨道，使权重都为NaN。我怀疑梯度计算不正确。不过，损失函数在评估过程中起作用。
是的，tf.linalg.svd可以用于损失函数，raw_opsSvd具有梯度，如tf.raw_ops文档页面上所示：
Op名称 具有渐变 Svd ✔️