双重或双层循环中tf.GradientTape的注意点

    科技2022-07-20  140

    最近在写RL的时候发现一个有意思的现象。话不多说上简化版的代码:

    import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' x = tf.constant(3.0) with tf.GradientTape(persistent = False)
    Processed: 0.010, SQL: 8