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可以实现相同的实现，因为它支持动态计算图