Browse Source

!10158 Remove Int16 Support for MultiFieldEmbedding For GPU Device

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

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

@@ -349,12 +349,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.


Loading…
Cancel
Save