grad_tape_ex1

def tf_gradient_tape(x):
    """
    Simple implementation to understand the functioning of gradient tape
    Inputs:
    x: Tensor value 

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