2022-02-19 15:27:46 +08:00
|
|
|
|
mindspore.ops.EmbeddingLookup
|
|
|
|
|
|
===============================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.ops.EmbeddingLookup
|
|
|
|
|
|
|
|
|
|
|
|
根据指定的索引，返回输入Tensor的切片。
|
|
|
|
|
|
|
|
|
|
|
|
此算子在 `axis = 0` 上的运行与GatherV2的功能相似，只是多一个输入： `offset` 参数。
|
|
|
|
|
|
|
|
|
|
|
|
**输入：**
|
|
|
|
|
|
|
2022-03-10 11:26:08 +08:00
|
|
|
|
- **input_params** (Tensor) - shape为 :math:`(x_1, x_2, ..., x_R)` 的Tensor。是一个Tensor切片。当前，只支持二维。
|
|
|
|
|
|
- **input_indices** (Tensor) - shape为 :math:`(y_1, y_2, ..., y_S)` 的Tensor。指定输入Tensor元素的索引。当取值超出 `input_params` 的范围时，超出部分在输出时填充为0。不支持负值，如果为负值，则结果未定义。其数据类型为int32或int64。
|
2022-02-19 15:27:46 +08:00
|
|
|
|
- **offset** (int) - 指定此 `input_params` 切片的偏移值。因此，实际索引等于 `input_indices` 减去 `offset` 。
|
|
|
|
|
|
|
|
|
|
|
|
**输出：**
|
|
|
|
|
|
|
|
|
|
|
|
Tensor，shape为 :math:`(z_1, z_2, ..., z_N)` 的Tensor。数据类型与 `input_params` 相同。
|
|
|
|
|
|
|
|
|
|
|
|
**异常：**
|
|
|
|
|
|
|
|
|
|
|
|
- **TypeError** - `input_indices` 的数据类型不是int。
|
|
|
|
|
|
- **ValueError** - `input_params` 的shape长度大于2。
|
