forked from mindspore-Ecosystem/mindspore
22 lines
647 B
ReStructuredText
22 lines
647 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类型。
|