tf.keras.initializers.Identity

Initializer that generates the identity matrix.

Inherits From: Initializer

Also available via the shortcut function tf.keras.initializers.identity.

Only usable for generating 2D matrices.

Examples:

# Standalone usage:
initializer = tf.keras.initializers.Identity()
values = initializer(shape=(2, 2))
# Usage in a Keras layer:
initializer = tf.keras.initializers.Identity()
layer = tf.keras.layers.Dense(3, kernel_initializer=initializer)

gain Multiplicative factor to apply to the identity matrix.

Methods

from_config

View source

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, the output of get_config.

Returns
A tf.keras.initializers.Initializer instance.

get_config

View source

Returns the initializer's configuration as a JSON-serializable dict.

Returns
A JSON-serializable Python dict.

__call__

View source

Returns a tensor object initialized to a 2D identity matrix.

Args
shape Shape of the tensor. It should have exactly rank 2.
dtype Optional dtype of the tensor. Only floating point types are supported. If not specified, tf.keras.backend.floatx() is used, which default to float32 unless you configured it otherwise (via tf.keras.backend.set_floatx(float_dtype))
**kwargs Additional keyword arguments.