SoftmaxFusionPass

融合模式

该融合规则将SoftmaxV2输入和输出添加TransposeD算子。

融合为

使用约束

• 输入shape约束：

  输入仅支持三维或者五维。

  • 当输入是三维时，输入shape最后两维度大小必须是{8732, 21}。
  • 当输入是五维时，输入shape最后四维度大小必须是{224, 224, 160, 4}。
• 输入参数的大小约束：

  输入参数的大小（元素个数×dtype大小）不要超过int32的最大值：2^32-1。

• 输入属性约束：

  仅支持属性axes的第一个值指定最后一根轴。

支持的型号

Atlas 训练系列产品