具有多个输出的TensorFlow自定义渐变

时间:2020-03-09 22:31:40

标签: tensorflow

如何为具有多个输出的现有tensorflow方法编写自定义梯度?例如,在计算行列式时,我想覆盖logName="projects/[project-name]/logs/events" resource.type="k8s_cluster" resource.labels.location="us-east1" resource.labels.cluster_name="production" 的梯度,而tf.linalg.slogdet返回张量的元组,它是行列式的符号和行列式绝对值的对数。

我做过类似的事情

slogdet

这给我一个错误

grad()接受1个位置参数,但给出2个位置

0 个答案:

没有答案