(beta)torch_npu.npu_indexing
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 推理系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
使用“begin,end,strides”数组对index结果进行计数。
函数原型
[object Object]
参数说明
- self (
Tensor):必选参数,输入张量。 - begin (
List[int]):必选参数,待选择的第一个值的index。 - end (
List[int]):必选参数,待选择的最后一个值的index。 - strides (
List[int]):必选参数,index增量。 - begin_mask (
int):可选参数,位掩码(bitmask),其中位“i”为“1”意味着忽略开始值,尽可能使用最大间隔。默认值为0。 - end_mask (
int):可选参数,类似于“begin_mask”。默认值为0。 - ellipsis_mask (
int):可选参数,位掩码,其中位“i”为“1”意味着第“i”个位置实际上是省略号。默认值为0。 - new_axis_mask (
int):可选参数,位掩码,其中位“i”为“1”意味着在第“i”位创建新的1D shape。默认值为0。 - shrink_axis_mask (
int):可选参数,位掩码,其中位“i”意味着第“i”位应缩小维数。默认值为0。
调用示例
[object Object]