FilesLast commitLast update
13 days ago
14 days ago
3 months ago
12 days ago
14 days ago
14 days ago
14 days ago
14 days ago
14 days ago
14 days ago
14 days ago
README

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算子。