文件最后提交记录最后更新时间
25 天前
25 天前
7 个月前
2 个月前
2 个月前
2 个月前
8 个月前
1 个月前
README

ForeachZeroInplace

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT ×
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×
Kirin X90 处理器系列产品
Kirin 9030 处理器系列产品

功能说明

  • 算子功能:原地更新输入张量列表,输入张量列表的每个张量置为0。

  • 计算公式:

    x=[x0,x1,...xn−1]x = [{x_0}, {x_1}, ... {x_{n-1}}]\\

    xi=0(i=0,1,...n−1)x_i = 0 (i=0,1,...n-1)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入/输出 表示进行置0运算的输入和输出张量列表,对应公式中的`x`。该参数中所有Tensor的数据类型保持一致。出参`x`的数据类型、数据格式和shape与入参`x`的数据类型、数据格式和shape一致。 FLOAT32、FLOAT16、INT32、INT16、BFLOAT16 ND
  • Kirin X90/Kirin 9030处理器系列产品:不支持BFLOAT16。

约束说明

  • inplace操作的输入/输出不支持非连续Tensor。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_foreach_zero_inplace 通过aclnnForeachZeroInplace接口方式调用ForeachZeroInplace算子。
图模式 - 通过算子IR构图方式调用ForeachZeroInplace算子。