ShiftLeft（左移位数为Tensor）

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

产品支持情况

功能说明

对源操作数中的每个元素进行左移操作。根据源操作数的数据类型，左移操作分为以下两种情况：

• 数据类型为无符号类型：执行逻辑左移。逻辑左移会将二进制数整体向左移动指定的位数，最高位被丢弃，最低位用0填充。例如，二进制数1010101010101010（uint16_t 类型）逻辑左移1位后，结果为0101010101010100。
• 数据类型为有符号类型：执行算术左移。算术左移会将二进制数整体向左移动指定的位数，次高位被丢弃，最低位用0填充。例如，二进制数1010101010101010（int16_t 类型）算术左移1位后，结果为1101010101010100；算术左移3位后，结果为1101010101010000。

函数原型

template <typename T, typename U>
__aicore__ inline void ShiftLeft(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::ShiftLeft(dstLocal, srcLocal0, srcLocal1, 512);

结果示例如下：

输入数据srcLocal0：[1 2 3 ... 512]
输入数据srcLocal1：[2 2 2 ... 2]
输出数据dstLocal：[4 8 12 ... 2048]

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