Improve the docstring for the tf.keras.initializer.get()
PiperOrigin-RevId: 363753200
Change-Id: I0d489400ebf5e396390653de256aa38f0e9c4290
diff --git a/tensorflow/python/keras/initializers/__init__.py b/tensorflow/python/keras/initializers/__init__.py
index b610097..6c415b8 100644
--- a/tensorflow/python/keras/initializers/__init__.py
+++ b/tensorflow/python/keras/initializers/__init__.py
@@ -148,6 +148,38 @@
@keras_export('keras.initializers.get')
def get(identifier):
+ """Retrieve a Keras initializer by the identifier.
+
+ The `identifier` may be the string name of a initializers function or class (
+ case-sensitively).
+
+ >>> identifier = 'Ones'
+ >>> tf.keras.initializers.deserialize(identifier)
+ <...tensorflow.python.keras.initializers.initializers_v2.Ones...>
+
+ You can also specify `config` of the initializer to this function by passing
+ dict containing `class_name` and `config` as an identifier. Also note that the
+ `class_name` must map to a `Initializer` class.
+
+ >>> cfg = {'class_name': 'Ones', 'config': {}}
+ >>> tf.keras.initializers.deserialize(cfg)
+ <...tensorflow.python.keras.initializers.initializers_v2.Ones...>
+
+ In the case that the `identifier` is a class, this method will return a new
+ instance of the class by its constructor.
+
+ Args:
+ identifier: String or dict that contains the initializer name or
+ configurations.
+
+ Returns:
+ Initializer instance base on the input identifier.
+
+ Raises:
+ ValueError: If the input identifier is not a supported type or in a bad
+ format.
+ """
+
if identifier is None:
return None
if isinstance(identifier, dict):