grad_tape_2_corr

def tf_gradient_tape2(x):
    """
    Simple implementation to understand the functioning of gradient tape for chain rule
    Inputs:
    x: Tensor value 

    Returns:
    EagerTensor: Derivative of z with respect to input tensor x
    """
    with tf.GradientTape() as t:
        t.watch(x)
        y=x*x  ## Defining y(x)=x**2
        z=y*y  ## Defining z(y)=y**2
    # Calculating the derivative of z with respect to x
    dz_dx = t.gradient(z, x)
    return dz_dx