mindspore.experimental.es.ESEmbeddingSmallTableLookup

class mindspore.experimental.es.ESEmbeddingSmallTableLookup(name, rank_id, rank_size, small_table_to_variable)[源代码]

提供小表特征值keys的查询功能。

警告

这是一个实验性的EmbeddingService接口。

参数：

  • name (str) - 小表名字。

  • rank_id (int) - 查询小表key时的rank id。

  • rank_size (int) - 查询小表是的rank size。

  • small_table_to_variable (dict[str, parameter]) - 存储小表信息的dict：key为小表名称，value为小表对应的参数信息。

输入：

  • ids_list (Tensor) - 小表的每个特征keys。

支持平台：

Atlas A2 training series products