tf.keras.initializers.Identity

View source on GitHub

Initializer that generates the identity matrix.

Inherits From: Initializer

tf.keras.initializers.Identity(
    gain=1.0
)

Only use for 2D matrices.

Args:

• gain: Multiplicative factor to apply to the identity matrix.

Methods

__call__

View source

__call__(
    shape, dtype=tf.dtypes.float32
)

Returns a tensor object initialized as specified by the initializer.

Args:

• shape: Shape of the tensor.
• dtype: Optional dtype of the tensor. Only floating point types are supported.

Raises:

• ValueError: If the dtype is not floating point

from_config

View source

@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 of get_config.

Returns:

An Initializer instance.

get_config

View source

get_config()

Returns the configuration of the initializer as a JSON-serializable dict.

Returns:

A JSON-serializable Python dict.