# Importing the library
import tensorflow as tf
 
# Initializing the Tensor
x = tf.Variable(2.0, name ="x")
z = tf.Variable(4.0, name ="z")
 
with tf.GradientTape() as gfg:
  # y will evaluate to 16.0 i.e 4**2
  y = tf.grad_pass_through(x.assign)(z**2)
 
# res will evaluate to 8.0
res = gfg.gradient(y, z)
 
# Printing result
print("y: ", y)
print("res: ", res)

y:  tf.Tensor(16.0, shape=(), dtype=float32)
res:  tf.Tensor(8.0, shape=(), dtype=float32)

# Importing the library
import tensorflow as tf
 
# Initializing the Tensor
x = tf.Variable(3.0, name ="x")
 
with tf.GradientTape() as gfg:
  # y will evaluate to 9.0 i.e 3**2
  y = tf.grad_pass_through(x.assign)(x**2)
 
# res will evaluate to 6.0
res = gfg.gradient(y, x)
 
# Printing result
print("y: ", y)
print("res: ", res)

y:  tf.Tensor(9.0, shape=(), dtype=float32)
res:  tf.Tensor(6.0, shape=(), dtype=float32)