def model(a, b): with tape_lib.GradientTape() as tape: tape.watch(a) tape.watch(b) result = unified_math_ops.add(a, b) grads = tape.gradient(result, [a, b]) return grads
def model(a): with tape_lib.GradientTape() as tape: tape.watch(a) result = unified_math_ops.log1p(a) grads = tape.gradient(result, a) return grads
def model(t): with tape_lib.GradientTape() as tape: tape.watch(t) result = unified_nn_ops.relu(t) grads = tape.gradient(result, t) return grads