tf.function使用所有CPU RAM

Question

我不明白为什么下面发布的功能会耗尽我的所有RAM。我能理解我是否在急切地运行它，但是我认为tf.function的目的是创建一个可重用的图形，就像创建一个操作并在tf 1.x中运行它一样。我是tensorflow 2.0的新手，所以我可能对tf.function正在执行的操作有错误的认识。

@tf.function
def clip_w(self, weight):
    return tf.clip_by_value(weight, -0.01, 0.01)

有人可以帮助我理解吗？谢谢

编辑：这是我使用此功能的代码

def clip_weights(self):
        for l in self.C.layers:
            weights = l.get_weights()
            weights = [self.clip_w(w) for w in weights]
            l.set_weights(weights)