tf.clip_by_global_norm
View source on GitHub |
Clips values of multiple tensors by the ratio of the sum of their norms.
tf.clip_by_global_norm( t_list, clip_norm, use_norm=None, name=None )
Given a tuple or list of tensors t_list
, and a clipping ratio clip_norm
, this operation returns a list of clipped tensors list_clipped
and the global norm (global_norm
) of all tensors in t_list
. Optionally, if you've already computed the global norm for t_list
, you can specify the global norm with use_norm
.
To perform the clipping, the values t_list[i]
are set to:
t_list[i] * clip_norm / max(global_norm, clip_norm)
where:
global_norm = sqrt(sum([l2norm(t)**2 for t in t_list]))
If clip_norm > global_norm
then the entries in t_list
remain as they are, otherwise they're all shrunk by the global ratio.
If global_norm == infinity
then the entries in t_list
are all set to NaN
to signal that an error occurred.
Any of the entries of t_list
that are of type None
are ignored.
This is the correct way to perform gradient clipping (Pascanu et al., 2012).
However, it is slower than clip_by_norm()
because all the parameters must be ready before the clipping operation can be performed.
Args | |
---|---|
t_list | A tuple or list of mixed Tensors , IndexedSlices , or None. |
clip_norm | A 0-D (scalar) Tensor > 0. The clipping ratio. |
use_norm | A 0-D (scalar) Tensor of type float (optional). The global norm to use. If not provided, global_norm() is used to compute the norm. |
name | A name for the operation (optional). |
Returns | |
---|---|
list_clipped | A list of Tensors of the same type as list_t . |
global_norm | A 0-D (scalar) Tensor representing the global norm. |
Raises | |
---|---|
TypeError | If t_list is not a sequence. |
References:
On the difficulty of training Recurrent Neural Networks: Pascanu et al., 2012 (pdf)
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/clip_by_global_norm