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