| @@ -145,7 +145,9 @@ class EmbeddingLookup(Cell): | |||||
| Args: | Args: | ||||
| vocab_size (int): Size of the dictionary of embeddings. | vocab_size (int): Size of the dictionary of embeddings. | ||||
| embedding_size (int): The size of each embedding vector. | embedding_size (int): The size of each embedding vector. | ||||
| param_init (str): The initialize way of embedding table. Default: 'normal'. | |||||
| param_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the embedding_table. | |||||
| Refer to class `initializer` for the values of string when a string | |||||
| is specified. Default: 'normal'. | |||||
| target (str): Specifies the target where the op is executed. The value must in | target (str): Specifies the target where the op is executed. The value must in | ||||
| ['DEVICE', 'CPU']. Default: 'CPU'. | ['DEVICE', 'CPU']. Default: 'CPU'. | ||||
| slice_mode (str): The slicing way in semi_auto_parallel/auto_parallel. The value must get through | slice_mode (str): The slicing way in semi_auto_parallel/auto_parallel. The value must get through | ||||
| @@ -291,7 +293,9 @@ class MultiFieldEmbeddingLookup(EmbeddingLookup): | |||||
| vocab_size (int): The size of the dictionary of embeddings. | vocab_size (int): The size of the dictionary of embeddings. | ||||
| embedding_size (int): The size of each embedding vector. | embedding_size (int): The size of each embedding vector. | ||||
| field_size (int): The field size of the final outputs. | field_size (int): The field size of the final outputs. | ||||
| param_init (str): The initialize way of embedding table. Default: 'normal'. | |||||
| param_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the embedding_table. | |||||
| Refer to class `initializer` for the values of string when a string | |||||
| is specified. Default: 'normal'. | |||||
| target (str): Specifies the target where the op is executed. The value must in | target (str): Specifies the target where the op is executed. The value must in | ||||
| ['DEVICE', 'CPU']. Default: 'CPU'. | ['DEVICE', 'CPU']. Default: 'CPU'. | ||||
| slice_mode (str): The slicing way in semi_auto_parallel/auto_parallel. The value must get through | slice_mode (str): The slicing way in semi_auto_parallel/auto_parallel. The value must get through | ||||