Fix typo in docstring for custom_gradient.
PiperOrigin-RevId: 294920338 Change-Id: Id176dc2a67c4f781fec358b8c588a4b747adb2b2
This commit is contained in:
parent
d545a37ebd
commit
41313e71ac
@ -100,8 +100,8 @@ def custom_gradient(f=None):
|
||||
return tf.math.log(1 + tf.exp(x))
|
||||
```
|
||||
|
||||
Due to numerical instability, the gradient this function evaluated at x=100 is
|
||||
NaN. For example:
|
||||
Due to numerical instability, the gradient of this function evaluated at x=100
|
||||
is NaN. For example:
|
||||
|
||||
```python
|
||||
x = tf.constant(100.)
|
||||
|
Loading…
x
Reference in New Issue
Block a user