当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。