ShiftLeft（左移位数为Tensor）
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
x |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
对源操作数中的每个元素进行左移操作。根据源操作数的数据类型，左移操作分为以下两种情况：
- 数据类型为无符号类型：执行逻辑左移。逻辑左移会将二进制数整体向左移动指定的位数，最高位被丢弃，最低位用0填充。例如，二进制数1010101010101010（uint16_t 类型）逻辑左移1位后，结果为0101010101010100。
- 数据类型为有符号类型：执行算术左移。算术左移会将二进制数整体向左移动指定的位数，次高位被丢弃，最低位用0填充。例如，二进制数1010101010101010（int16_t 类型）算术左移1位后，结果为1101010101010100；算术左移3位后，结果为1101010101010000。
函数原型
1 2 | 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) |
参数说明
参数名 |
描述 |
|---|---|
T |
源/目的操作数数据类型。 Atlas 350 加速卡，支持的数据类型为：int8_t、uint8_t、int16_t、uint16_t、int32_t、uint32_t、int64_t、uint64_t。 |
U |
源操作数数据类型。 Atlas 350 加速卡，支持的数据类型为：int8_t、int16_t、int32_t、int64_t。 |
参数名称 |
输入/输出 |
说明 |
|---|---|---|
dst |
输出 |
目的操作数。 类型为LocalTensor，支持的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的起始地址需要32字节对齐。 |
src0 |
输入 |
源操作数。 类型为LocalTensor，支持的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的起始地址需要32字节对齐。 数据类型需要与目的操作数保持一致。 |
src1 |
输入 |
存放左移位数的LocalTensor，数据类型的字节数需要与源src0操作数Tensor中的元素数据类型的字节数相匹配，不支持设置为负数。 |
count |
输入 |
参与计算的元素个数。 |
返回值说明
无
约束说明
- 操作数地址对齐要求请参见通用地址对齐约束。
- 对于逻辑位移（无符号数据类型），如果位移量大于数据类型位宽，则输出为0。
- 对于算数位移（有符号数据类型），如果src0小于0，src1小于0，并且位移量大于数据类型位宽，则输出-1；如果src0大于0，并且位移量大于数据类型位宽，则输出0。
调用示例
1 | AscendC::ShiftLeft(dstLocal, srcLocal0, srcLocal1, 512); |
结果示例如下：
输入数据srcLocal0：[1 2 3 ... 512] 输入数据srcLocal1：[2 2 2 ... 2] 输出数据dstLocal：[4 8 12 ... 2048]
父主题： 逻辑计算