LeftShift

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
Atlas A2系列产品 √
Atlas 200I/500 A2推理产品 ×
Atlas推理系列产品 ×
Atlas训练系列产品 ×

功能说明

  • 算子功能:按元素计算输入张量x与y的按位左移操作。

  • 计算公式:

zi=xi<<yiz_i = x_i << y_i

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入x是一个k维张量,是待进行left_shift计算的入参,公式中的x_i。 INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
y 输入 待进行left_shift计算的入参,公式中的y_i。 INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND
z 输出 待进行left_shift计算的出参,公式中的z_i。 INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 ND

约束说明

当shiftBits为负数或不小于x数据类型的位宽时,输出为0,与PyTorch的torch.bitwise_left_shift保持一致。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_left_shift 通过aclnnLeftShift接口方式调用left_shift算子。
aclnn调用 test_aclnn_left_shifts 通过aclnnLeftShifts接口方式调用left_shift算子(shiftBits为标量)。
图模式调用 test_geir_left_shift 通过算子IR构图方式调用LeftShift算子。