From 8e09880686ac64b722c645c76412414c740c6a90 Mon Sep 17 00:00:00 2001 From: huangxinjing Date: Fri, 18 Dec 2020 11:02:22 +0800 Subject: [PATCH] Fix doc --- mindspore/nn/layer/embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/nn/layer/embedding.py b/mindspore/nn/layer/embedding.py index d50cc912ab..0e95864880 100755 --- a/mindspore/nn/layer/embedding.py +++ b/mindspore/nn/layer/embedding.py @@ -328,12 +328,12 @@ class MultiFieldEmbeddingLookup(EmbeddingLookup): Inputs: - **input_indices** (Tensor) - The shape of tensor is :math:`(batch_size, seq_length)`. Specifies the indices of elements of the original Tensor. Input_indices must be a 2d tensor in - this interface. Type is Int16, Int32, Int64. + this interface. Type is Int32, Int64. - **input_values** (Tensor) - The shape of tensor is :math:`(batch_size, seq_length)`. Specifies the weights of elements of the input_indices. The lookout vector will multiply with the input_values. Type is Float32. - **field_ids** (Tensor) - The shape of tensor is :math:`(batch_size, seq_length)`. - Specifies the field id of elements of the input_indices. Type is Int16, Int32. + Specifies the field id of elements of the input_indices. Type is Int32. Outputs: Tensor, the shape of tensor is :math:`(batch_size, field_size, embedding_size)`. Type is Float32.