|
|
|
@@ -128,8 +128,10 @@ class EmbeddingLookup(Cell): |
|
|
|
vocab_size (int): Size of the dictionary of embeddings. |
|
|
|
embedding_size (int): The size of each embedding vector. |
|
|
|
param_init (str): The initialize way of embedding table. Default: 'normal'. |
|
|
|
target (str): Specify the target where the op is executed. Default: 'CPU'. |
|
|
|
slice_mode (str): The slicing way in semi auto parallel/auto parallel. Default: 'batch_slice'. |
|
|
|
target (str): Specify the target where the op is executed. The value should in |
|
|
|
['DEVICE', 'CPU']. Default: 'CPU'. |
|
|
|
slice_mode (str): The slicing way in semi auto parallel/auto parallel. The value should get through |
|
|
|
nn.EmbeddingLookUpSplitMode. Default: 'batch_slice'. |
|
|
|
manual_shapes (tuple): The accompaniment array in field slice mode. |
|
|
|
|
|
|
|
Inputs: |
|
|
|
|