mindspore.experimental.es.ESEmbeddingSmallTableLookup

View Source On Gitee
class mindspore.experimental.es.ESEmbeddingSmallTableLookup(name, rank_id, rank_size, small_table_to_variable)[source]

Look up a data_parallel embedding.

Warning

This is an experimental EmbeddingService API that is subject to change.

Parameters

  • name (str) – The data_parallel embedding name.

  • rank_id (int) – The rank id when look up data_parallel embedding key.

  • rank_size (int) – The rank size when look up data_parallel embedding key.

  • small_table_to_variable (dict[str, parameter]) – The dict to restore data_parallel embedding information: key is table name, value is parameter.

Inputs:

  • ids_list (Tensor) - The keys of each feature in data_parallel embedding.

Supported Platforms:

Atlas A2 training series products