（beta）torch_npu.npu_indexing

函数原型

功能说明

使用“begin,end,strides”数组对index结果进行计数。

参数说明

• self (Tensor) - 输入张量。
• begin (ListInt) - 待选择的第一个值的index。
• end (ListInt) - 待选择的最后一个值的index。
• strides (ListInt) - index增量。
• begin_mask (Int，默认值为0) - 位掩码（bitmask），其中位“i”为“1”意味着忽略开始值，尽可能使用最大间隔。
• end_mask (Int，默认值为0) - 类似于“begin_mask”。
• ellipsis_mask (Int，默认值为0) - 位掩码，其中位“i”为“1”意味着第“i”个位置实际上是省略号。
• new_axis_mask (Int，默认值为0) - 位掩码，其中位“i”为“1”意味着在第“i”位创建新的1D shape。
• shrink_axis_mask (Int，默认值为0) - 位掩码，其中位“i”意味着第“i”位应缩小维数。

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]

调用示例