tensorflow - 在 Keras 中，是否有描述字符串名称到初始化程序、优化程序等的类映射的文档？

Question

是否有任何文档描述了哪些字符串名称映射到 Keras 中的哪些对象？例如，下面我从 tf.keras.layers 创建了一个嵌入层我可以使用 'uniform'映射到 tf.keras.initializers.RandomUniform类(class)。

tf.keras.layers.Embedding(1000, 64, embeddings_initializer='uniform')

但我只能通过查看该用法的示例才知道这一点。我认为支持的字符串形式在某处记录，但我似乎无法找到这样的文档，并且挖掘代码太抽象而无法轻松理解。

版本:TF 1.13.1

Answer 1

在 TF 的 keras 实现中没有可用的字符串常量列表(我想，在原始 keras 中也没有)。

对于 initializer案例'uniform' string 转换为 config 并在该配置上调用结构方法，并提示从 initializers 命名空间创建对象(可以在此处找到 def deserialize_keras_object ):

config = {'class_name': str(identifier), 'config': {}}

deserialize_keras_object(
      config,
      module_objects=globals(),
      custom_objects=custom_objects,
      printable_module_name='initializer')

因此，我想不出比以下更好的方法来列出所有初始化程序:

import tensorflow as tf

for k, v in tf.keras.initializers.__dict__.items():
    if not k[0].isupper() and not k[0] == "_":
        print(k)

和输出，虽然有额外的值，就像:

constant
glorot_normal
glorot_uniform
identity
ones
orthogonal
zeros
he_normal
he_uniform
lecun_normal
lecun_uniform
normal
random_normal
random_uniform
uniform
truncated_normal
deserialize
get
serialize