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

ThresholdGradV2D

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品

功能说明

  • 算子功能:求thresholdv2d函数梯度。
  • 计算公式:

y={input_gradient,input_feature>threshold0,input_feature<=threshold y = \begin{cases} input\_ gradient, & input\_ feature > threshold \\ 0, & input\_ feature <= threshold \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
input_gradient 输入 反向传播梯度 FLOAT、FLOAT16、BFLOAT16、INT32、INT8、UINT8 ND
input_feature 输入 与"input_gradient"具有相同的类型、格式和形状。 FLOAT、FLOAT16、BFLOAT16、INT32、INT8、UINT8 ND
threshold 属性 FLOAT
output_backprops 输出 公式中的输出张量 FLOAT、FLOAT16、BFLOAT16、INT32、INT8、UINT8 ND

调用说明

调用方式 样例代码 说明
Aclnn模式 test_aclnn_threshold_backward.cpp 通过Aclnn接口调用ThresholdGradV2D算子。

约束说明

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
ilovescrapy 个人开发者 ThresholdGradV2D 2026/3/23 ThresholdGradV2D算子适配开源仓