/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.EmbeddingLookup.rst
20 строк
1 KB
zhangyi
modify the format of API files
25 июл 2022, 10:26
25 июл 2022, 10:26
46e8f6a
Код
Авторство
О чём код?
mindspore.ops.EmbeddingLookup =============================== .. py:class:: mindspore.ops.EmbeddingLookup 根据指定的索引，返回输入Tensor的切片。 此算子在 `axis = 0` 上的运行与GatherV2的功能相似，只是多一个 `offset` 输入。 输入： - **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。 - **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。