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