文件最后提交记录最后更新时间
4 个月前
8 个月前
8 个月前
5 个月前
5 个月前
8 个月前
8 个月前
4 个月前
README

RepeatInterleaveGrad

产品支持情况

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

功能说明

  • 算子功能:算子RepeatInterleave的反向, 将y_grad tensor的axis维度按repeats进行ReduceSum。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
y_grad 输入 表示待被ReduceSum的输入tensor。 FLOAT、FLOAT16、BFLOAT16 ND
repeats 输入 表示重复的次数。 INT32、INT64 ND
x_grad 输出 表示ReduceSum完成的输出tensor。 FLOAT、INT32 ND
axis 可选属性
  • 表示ReduceSum作用的维度。取值范围为[-n, n),其中n为y_grad的维度。
  • 默认值为-1。
int64_t -

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_repeat_interleave_grad 通过aclnnRepeatInterleaveGrad接口方式调用RepeatInterleaveGrad算子。
图模式调用 test_geir_repeat_interleave_grad 通过算子IR构图方式调用RepeatInterleaveGrad算子。