tf.compat.v1.initializers.identity | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.initializers.identity
Initializer that generates the identity matrix.
tf.compat.v1.initializers.identity(
gain=1.0,
dtype=tf.dtypes.float32
)
Only use for 2D matrices.
| Args | |
|---|---|
| gain | Multiplicative factor to apply to the identity matrix. |
| dtype | Default data type, used if no dtype argument is provided when calling the initializer. Only floating point types are supported. |
Methods
from_config
@classmethod
from_config( config )
Instantiates an initializer from a configuration dictionary.
Example:
initializer = RandomUniform(-1, 1)
config = initializer.get_config()
initializer = RandomUniform.from_config(config)
| Args | |
|---|---|
| config | A Python dictionary. It will typically be the output ofget_config. |
| Returns |
|---|
| An Initializer instance. |
get_config
get_config()
Returns the configuration of the initializer as a JSON-serializable dict.
| Returns |
|---|
| A JSON-serializable Python dict. |
__call__
__call__(
shape, dtype=None, partition_info=None
)
Returns a tensor object initialized as specified by the initializer.
| Args | |
|---|---|
| shape | Shape of the tensor. |
| dtype | Optional dtype of the tensor. If not provided use the initializer dtype. |
| partition_info | Optional information about the possible partitioning of a tensor. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.