文档反馈

问题文档片段

问题文档片段包含公式时，显示为空格。

提交类型

issue

有点复杂...

找人问问吧。

请选择提交类型

问题类型

规范和低错类

- 规范和低错类：

- 错别字或拼写错误，标点符号使用错误、公式错误或显示异常。

- 链接错误、空单元格、格式错误。

- 英文中包含中文字符。

- 界面和描述不一致，但不影响操作。

- 表述不通顺，但不影响理解。

- 版本号不匹配：如软件包名称、界面版本号。

易用性

- 易用性：

- 关键步骤错误或缺失，无法指导用户完成任务。

- 缺少主要功能描述、关键词解释、必要前提条件、注意事项等。

- 描述内容存在歧义指代不明、上下文矛盾。

- 逻辑不清晰，该分类、分项、分步骤的没有给出。

正确性

- 正确性：

- 技术原理、功能、支持平台、参数类型、异常报错等描述和软件实现不一致。

- 原理图、架构图等存在错误。

- 命令、命令参数等错误。

- 代码片段错误。

- 命令无法完成对应功能。

- 界面错误，无法指导操作。

- 代码样例运行报错、运行结果不符。

风险提示

- 风险提示：

- 对重要数据或系统存在风险的操作，缺少安全提示。

内容合规

- 内容合规：

- 违反法律法规，涉及政治、领土主权等敏感词。

- 内容侵权。

请选择问题类型

问题描述

点击输入详细问题描述，以帮助我们快速定位问题。

提交

文档反馈

mindspore.nn.EmbeddingLookup

class mindspore.nn.EmbeddingLookup(vocab_size, embedding_size, param_init='normal', target='CPU', slice_mode='batch_slice', manual_shapes=None, max_norm=None, sparse=True, vocab_cache_size=0)

嵌入查找层。

与嵌入层功能相同，主要用于自动并行或半自动并行时，存在大规模嵌入层的异构并行场景。

说明

当’target’设置为’CPU’时，此模块将使用ops.EmbeddingLookup().set_device(‘CPU’)，在lookup表指定了’offset = 0’。 当’target’设置为’DEVICE’时，此模块将使用ops.Gather()，在lookup表指定了’axis = 0’。 在字段切片模式下，必须指定manual_shapes。此tuple包含vocab[i]元素, vocab[i]是第i部分的行号。

参数：

• vocab_size (int) - 嵌入词典的大小。

• embedding_size (int) - 每个嵌入向量的大小。

• param_init (Union[Tensor, str, Initializer, numbers.Number]) - embedding_table的初始化方法。当指定为字符串，字符串取值请参见类 Initializer 。默认值：’normal’。

• target (str) - 指定执行操作的’target’。取值范围为[‘DEVICE’, ‘CPU’]。默认值：’CPU’。

• slice_mode (str) - semi_auto_parallel或auto_parallel模式下的切片方式。该值必须通过 nn.EmbeddingLookup 获得。默认值：’nn.EmbeddingLookup.BATCH_SLICE’。

• manual_shapes (tuple) - 字段切片模式下的伴随数组（accompaniment array），默认值：None。

• max_norm (Union[float, None]) - 最大剪切值。数据类型必须为float16、float32或None。默认值：None。

• sparse (bool) - 使用稀疏模式。当’target’设置为’CPU’时，’sparse’必须为True。默认值：True。

• vocab_cache_size (int) - 嵌入字典的缓存大小。默认值：0。仅在训练模式和’DEVICE’目标中有效。相应优化器的力矩参数也将设置为缓存大小。此外需注意，它还会消耗’DEVICE’内存，因此建议合理设置参数值，避免内存不足。

输入：

• input_indices (Tensor) - shape为 $(y_1,y_2,...,y_S)$ 的Tensor。指定原始Tensor元素的索引。当取值超出embedding_table的范围时，超出部分在输出中填充为0。不支持负值，如果为负值，则结果未定义。在semi auto parallel或auto parallel模式下运行时，Input_indices只能是此接口中的二维Tensor。

输出：

Tensor，shape为 $(z_1,z_2,...,z_N)$ 的Tensor。

异常：

• TypeError - vocab_size 、 embedding_size 或 vocab_cache_size 不是整数。

• TypeError - sparse 不是bool或 manual_shapes 不是tuple。

• ValueError - vocab_size 或 embedding_size 小于1。

• ValueError - vocab_cache_size 小于0。

• ValueError - target 既不是’CPU’也不是’DEVICE’。

• ValueError - slice_mode 不是’batch_slice’、’field_slice’、’table_row_slice’或’table_column_slice’。

• ValueError - sparse 为False且 target 为’CPU’。

• ValueError - slice_mode 为’field_slice’且 manual_shapes 是None。

支持平台：

Ascend GPU CPU

样例：

>>> input_indices = Tensor(np.array([[1, 0], [3, 2]]), mindspore.int32)
>>> result = nn.EmbeddingLookup(4,2)(input_indices)
>>> print(result.shape)
(2, 2, 2)