From f948bf95923a45e51318fe404e913da90f6896dc Mon Sep 17 00:00:00 2001 From: yao_yf Date: Thu, 19 Nov 2020 09:13:34 +0800 Subject: [PATCH] embeddinglookup has a omission --- mindspore/nn/layer/embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/embedding.py b/mindspore/nn/layer/embedding.py index e5b6e34fc7..6fd4bcedd7 100755 --- a/mindspore/nn/layer/embedding.py +++ b/mindspore/nn/layer/embedding.py @@ -248,7 +248,7 @@ class EmbeddingLookup(Cell): shp = self.shape(indices) + (self.embedding_size,) indices_flatten = self.reshape(indices, (-1,)) unique_id, unique_idx = self.unique(indices_flatten) - weight_unique = self.gatherv2(unique_id) + weight_unique = self.gatherv2(self.embedding_table, unique_id, 0) weight_flatten = self.gather_revert(weight_unique, unique_idx, 0) out = self.reshape(weight_flatten, shp) else: