CANN/asc-devkit ShiftRight右移Tensor

ShiftRight（右移位数为Tensor）

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

产品支持情况

功能说明

源操作数内每个元素做右移。

对无符号数据类型的源操作数做逻辑右移，对有符号数据类型的源操作数做算术右移。

逻辑右移为去掉最低位，最高位填充为0。

算术右移为去掉最低位，最高位复制符号位。

例：数据类型uint16_t，二进制数 1010101010101010，逻辑右移一位结果为 0101010101010101；

数据类型int16_t，二进制数 1010101010101010，算术右移一位结果为 1101010101010101；

数据类型int16_t，二进制数 1010101010101010，算术右移三位结果为 1111010101010101。

函数原型

template <typename T, typename U> __aicore__ inline void ShiftRight(const LocalTensor<T>& dst, const LocalTensor<T>& src0, const LocalTensor<U>& src1, const int32_t& count)

参数说明

表 1模板参数说明

表 2参数说明

返回值说明

无

约束说明

• 操作数地址对齐要求请参见通用地址对齐约束。

• 对于逻辑位移（无符号数据类型），如果位移量大于数据类型位宽，则输出为0。

• 对于算数位移（有符号数据类型），如果src0小于0，src1大于0，并且位移量大于数据类型位宽，则输出-1；如果src0大于0，并且位移量大于数据类型位宽，则输出0。

调用示例

AscendC::ShiftRight(dstLocal, srcLocal0, srcLocal1, 512);

结果示例如下：

输入数据(srcLocal0): [1 2 3 ... 512] 输入数据(srcLocal1): [2 2 2 ... 2] 输出数据(dstLocal): [0 0 0 1 1 1 1 ... 128]

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit