我想实现一个自定义损失,可以通过使用每个样本来计算。 损失的计算有点复杂,需要我为此使用外部 python 文件(或者可以假设我们将输入提供给函数)。
我该如何实现这个?
是否可以使用 @tf.function
注释并将其制成图表?
这就是它应该的样子
def loss(input,output):
loss = 0
for x, y in zip(input, output):
sim = Class(x)
a = sim.GetA()
b = sim.GetB()
loss = loss + np.linalg.norm(np.dot(a,b)+y)
return loss
最佳答案
通过 PyTorch 实现相同的功能是可能的,因为它支持动态计算图
关于python - Tensorflow 2.0.0 中的自定义损失,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58971511/