Tensorflow 2.0.0中的自定义损失

Question

我想实现一个自定义损失，可以通过使用每个样本来计算。 损耗的计算有些复杂，需要为此使用一个外部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

Answer 1

通过PyTorch可以实现相同的功能，因为它支持动态计算图