I would like to view updated gradient information after the optimization stage has been complete.
Using:
with tf.GradientTape() as tape: out = mode