Python tf.compat.v1.layers.experimental.set_keras_style用法及代碼示例

使用Keras-style 變量管理。

用法

tf.compat.v1.layers.experimental.set_keras_style()

啟用 keras 樣式後創建的所有 tf​​.layers 和 tf RNN 單元都使用Keras-style 變量管理。不允許使用 scope= 參數創建此類層，並且不允許使用 reuse=True。

此函數的目的是允許現有層的用戶在不破壞現有函數的情況下緩慢過渡到 Keras 層 API。

有關更多詳細信息，請參閱 keras_style_scope 的文檔。

請注意，一旦設置了 keras 樣式，它將為整個程序全局設置，並且不能取消設置。

例子：

set_keras_style()

model_1 = RNNModel(name="model_1")
model_2 = RNNModel(name="model_2")

# model_1 and model_2 are guaranteed to create their own variables.
output_1, next_state_1 = model_1(input, state)
output_2, next_state_2 = model_2(input, state)

assert len(model_1.weights) > 0
assert len(model_2.weights) > 0
assert(model_1.weights != model_2.weights)