Python tf.compat.v1.keras.initializers.RandomUniform用法及代码示例

生成具有均匀分布的张量的初始化程序。

继承自：random_uniform_initializer

用法

tf.compat.v1.keras.initializers.RandomUniform(
    minval=-0.05, maxval=0.05, seed=None, dtype=tf.dtypes.float32
)

迁移到 TF2

警告：这个 API 是为 TensorFlow v1 设计的。继续阅读有关如何从该 API 迁移到本机 TensorFlow v2 等效项的详细信息。见TensorFlow v1 到 TensorFlow v2 迁移指南有关如何迁移其余代码的说明。

虽然它是一个遗留的 compat.v1 api，但 tf.compat.v1.keras.initializers.RandomUniform 与即刻执行和 tf.function 兼容。

要切换到本机 TF2，请切换到使用 tf.keras.initializers.RandomUniform(不是从 compat.v1 )，如果需要更改默认 dtype，请使用 tf.keras.backend.set_floatx(float_dtype) 或在调用初始化程序时传递 dtype，而不是在构造初始化程序时传递它。

随机种子行为：

另请注意，如果您将种子传递给 TF2 初始化程序 API，它将在每次初始化时重用相同的种子(与 TF1 初始化程序不同)

到原生 TF2 的结构映射

前：

initializer = tf.compat.v1.keras.initializers.RandomUniform(
  minval=minval,
  maxval=maxval,
  seed=seed,
  dtype=dtype)

weight_one = tf.Variable(initializer(shape_one))
weight_two = tf.Variable(initializer(shape_two))

后：

initializer = tf.keras.initializers.RandomUniform(
  minval=minval,
  maxval=maxval,
  # seed=seed,  # Setting a seed in the native TF2 API
                # causes it to produce the same initializations
                # across multiple calls of the same initializer.
  )

weight_one = tf.Variable(initializer(shape_one, dtype=dtype))
weight_two = tf.Variable(initializer(shape_two, dtype=dtype))

如何映射参数

TF1 参数名称	TF2 参数名称	注意
minval	minval	没有更改默认值
maxval	maxval	没有更改默认值
seed	seed	不同的随机数生成语义(在未来版本中更改)。如果设置，TF2 版本将使用无状态随机数生成，即使在多次调用初始化程序实例时也会产生完全相同的初始化。 compat.v1 版本每次都会生成新的初始化。如果您每次都需要不同的初始化，请不要设置种子。相反，如果您需要确定性，可以使用 tf.random.set_seed 设置全局 tf 种子，或者使用单独的初始化程序实例和不同的种子来初始化每个权重。
dtype	dtype	TF2 原生 api 仅将其作为 __call__ arg，而不是构造函数 arg。
partition_info	-	(TF1 中的 __call__ arg)不支持

fixed-seed 行为差异示例

compat.v1 固定种子行为：

initializer = tf.compat.v1.keras.initializers.RandomUniform(seed=10)
a = initializer(shape=(2, 2))
b = initializer(shape=(2, 2))
tf.reduce_sum(a - b) == 0
<tf.Tensor:shape=(), dtype=bool, numpy=False>

后：

initializer = tf.keras.initializers.RandomUniform(seed=10)
a = initializer(shape=(2, 2))
b = initializer(shape=(2, 2))
tf.reduce_sum(a - b) == 0
<tf.Tensor:shape=(), dtype=bool, numpy=True>