使用tf.function的条件分支当使用带有条件分支的tf.function时,我在使用梯度带计算梯度时遇到问题 在梯度磁带范围内,我试图计算zw.r.tself.LMn的梯度。当我不使用@tf.function注释函数时,这工作得非 ...2024-09-30 已阅读: n次