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

生成初始化为 1 的张量的初始化程序。

用法

tf.compat.v1.keras.initializers.Ones(
    dtype=tf.dtypes.float32
)

迁移到 TF2

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

此 API 与 TF2 行为和 tf.function 兼容，并且可以使用 tf.keras.initializers.ones 立即迁移。

前：

>>> initializer = tf.compat.v1.keras.initializers.ones()
>>> initializer((1, 1))
<tf.Tensor:shape=(1, 1), dtype=float32, numpy=array([[1.]], dtype=float32)>

后：

>>> initializer = tf.keras.initializers.ones()
>>> initializer((1, 1))
<tf.Tensor:shape=(1, 1), dtype=float32, numpy=array([[1.]], dtype=float32)>