Python tf.GradientTape.stop_recording用法及代码示例

用法

@tf_contextlib.contextmanager
stop_recording()

暂时停止此磁带上的录制操作。

此上下文管理器处于活动状态时执行的操作不会记录在磁带上。这对于减少跟踪所有计算所使用的内存很有用。

例如：

x = tf.constant(4.0)
with tf.GradientTape() as tape:
  with tape.stop_recording():
    y = x ** 2
dy_dx = tape.gradient(y, x)
print(dy_dx)
None