forked from mindspore-Ecosystem/mindspore
19 lines
645 B
ReStructuredText
19 lines
645 B
ReStructuredText
mindspore.ops.SparseGatherV2
|
||
============================
|
||
|
||
.. py:class:: mindspore.ops.SparseGatherV2
|
||
|
||
基于指定的索引和axis返回输入Tensor的切片。
|
||
|
||
输入：
|
||
- **input_params** (Tensor) - 被切片的Tensor。shape： :math:`(x_1, x_2, ..., x_R)` 。
|
||
- **input_indices** (Tensor)- shape： :math:`(y_1, y_2, ..., y_S)` 。
|
||
指定切片的索引，取值须在 `[0, input_params.shape[axis])` 范围内。
|
||
- **axis** (int) - 进行索引的axis。
|
||
|
||
输出：
|
||
Tensor，shape： :math:`(z_1, z_2, ..., z_N)` 。
|
||
|
||
异常：
|
||
- **TypeError** - `axis` 不是int类型。
|