當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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

用法

@tf_contextlib.contextmanager
stop_recording()

生成(Yield)

  • None

拋出

  • RuntimeError 如果磁帶當前沒有錄製。

暫時停止此磁帶上的錄製操作。

此上下文管理器處於活動狀態時執行的操作不會記錄在磁帶上。這對於減少跟蹤所有計算所使用的內存很有用。

例如:

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

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.GradientTape.stop_recording。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。