View source on GitHub |
Initializer that generates the identity matrix.
Inherits From: Initializer
tf.keras.initializers.Identity(
gain=1.0
)
Only usable for generating 2D matrices.
Examples:
# Standalone usage:
initializer = Identity()
values = initializer(shape=(2, 2))
# Usage in a Keras layer:
initializer = Identity()
layer = Dense(3, kernel_initializer=initializer)
Args | |
---|---|
gain
|
Multiplicative factor to apply to the identity matrix. |
Methods
clone
clone()
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, the output of get_config() .
|
Returns | |
---|---|
An Initializer instance.
|
get_config
get_config()
Returns the initializer's configuration as a JSON-serializable dict.
Returns | |
---|---|
A JSON-serializable Python dict. |
__call__
__call__(
shape, dtype=None
)
Returns a tensor object initialized as specified by the initializer.
Args | |
---|---|
shape
|
Shape of the tensor. |
dtype
|
Optional dtype of the tensor. Only numeric or boolean dtypes
are supported. If not specified, keras.backend.floatx()
is used, which default to float32 unless you configured it
otherwise (via keras.backend.set_floatx(float_dtype) ).
|