Browse Source

!8365 add note for sparse in nn.EmbeddingLookup

From: @yao_yf
Reviewed-by: @stsuteng,@kisnwang
Signed-off-by: @stsuteng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
f4208a4a48
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/embedding.py

+ 1
- 1
mindspore/nn/layer/embedding.py View File

@@ -148,7 +148,7 @@ class EmbeddingLookup(Cell):
manual_shapes (tuple): The accompaniment array in field slice mode.
max_norm (Union[float, None]): A maximum clipping value. The data type must be float16, float32
or None. Default: None
sparse (bool): Using sparse mode. Currently, only support sparse mode when target is CPU. Default: True.
Inputs:
- **input_indices** (Tensor) - The shape of tensor is :math:`(y_1, y_2, ..., y_S)`.
Specifies the indices of elements of the original Tensor. Values can be out of range of embedding_table,


Loading…
Cancel
Save