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)