sigmoid_grad.py 236 B

1234567891011121314
  1. import tensorflow as tf
  2. a = tf.linspace(-10., 10., 10)
  3. with tf.GradientTape() as tape:
  4. tape.watch(a)
  5. y = tf.sigmoid(a)
  6. grads = tape.gradient(y, [a])
  7. print('x:', a.numpy())
  8. print('y:', y.numpy())
  9. print('grad:', grads[0].numpy())