Python tf.keras.initializers.Identity用法及代码示例

生成单位矩阵的初始化程序。

继承自：Initializer

用法

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

也可通过快捷函数 tf.keras.initializers.identity 获得。

仅可用于生成 2D 矩阵。

例子：

# 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)