From 9f443d2ee5e916267d7e1905b95ed6bb27685e5c Mon Sep 17 00:00:00 2001 From: ranqiu Date: Thu, 14 Sep 2017 16:42:13 +0800 Subject: [PATCH] Updating README.md according to the changes of external_memory --- mt_with_external_memory/README.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mt_with_external_memory/README.md b/mt_with_external_memory/README.md index 266ebbf9..507478e5 100644 --- a/mt_with_external_memory/README.md +++ b/mt_with_external_memory/README.md @@ -142,6 +142,7 @@ class ExternalMemory(object): name, mem_slot_size, boot_layer, + initial_weight, readonly=False, enable_interpolation=True): """ Initialization. @@ -154,6 +155,8 @@ class ExternalMemory(object): sequence layer has sequence length indicating the number of memory slots, and size as memory slot size. :type boot_layer: LayerOutput + :param initial_weight: Initializer for addressing weights. + :type initial_weight: LayerOutput :param readonly: If true, the memory is read-only, and write function cannot be called. Default is false. :type readonly: bool @@ -205,7 +208,7 @@ class ExternalMemory(object): - `_content_addressing`: 通过基于内容的寻址，计算得到读写操作的寻址强度。 - `_interpolation`: 通过插值寻址（当前寻址强度和上一时间步寻址强度的线性加权），更新当前寻址强度。 -- `_get_addressing_weight`: 调用上述两个寻址操作，获得对存储导员的读写操作的最终寻址强度。 +- `_get_addressing_weight`: 调用上述两个寻址操作，获得对存储单元的读写操作的最终寻址强度。 对外接口包含： @@ -214,6 +217,7 @@ class ExternalMemory(object): - 输入参数 `name`: 外部记忆单元名，不同实例的相同命名将共享同一外部记忆单元。 - 输入参数 `mem_slot_size`: 单个记忆槽（向量）的维度。 - 输入参数 `boot_layer`: 用于内存槽初始化的层。需为序列类型，序列长度表明记忆槽的数量。 + - 输入参数 `initial_weight`: 用于初始化寻址强度。 - 输入参数 `readonly`: 是否打开只读模式（例如打开只读模式，该实例可用于注意力机制）。打开只读模式，`write` 方法不可被调用。 - 输入参数 `enable_interpolation`: 是否允许插值寻址（例如当用于注意力机制时，需要关闭插值寻址）。 - `write`: 写操作。 -- GitLab