---
title: SoftmaxGradExtFusion
description: "该融合将符合下图左侧图结构的Mul、ReduceSumD、Sub这些小算子，融合成下图右侧的SoftmaxGradExt算子。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/maintenref/graphubfusionref/atlasrr_30_0128.html
sourcePath: /source/zh/canncommercial/900/maintenref/graphubfusionref/atlasrr_30_0128.html
indexId: 67d29aa8097d929786df26f148d79e8fbf7d5860e6a4c20df05774750f1d918f78
---
# SoftmaxGradExtFusion

#### 融合模式

该融合将符合下图左侧图结构的Mul、ReduceSumD、Sub这些小算子，融合成下图右侧的SoftmaxGradExt算子。


#### 使用约束

- 输入约束：

  - Mul_1节点的输入与Sub节点的第一个输入共用input0。
  - Mul_1节点的输入与Mul_2节点的输入共用input1。
  - ReduceSumD的axis参数必须为-1，keep_dims参数必须为True。
- 数据格式和shape约束：

  - input0、input1的shape必须是4D~8D。
  - input0和input1的shape最后两位必须是16，且数据格式为NZ。
  - input2的类型必须是scalar。
- 不支持动态shape场景。
- 数据类型约束：

  - input0、input1、input2的数据类型需要保持一致。
  - Atlas A2 训练系列产品 / Atlas A2 推理系列产品：数据类型支持FLOAT16、FLOAT32、BFLOAT16。


#### 支持的型号

Atlas A2 训练系列产品 / Atlas A2 推理系列产品