昇腾社区首页
中文
注册

void SetFirstTranspose(PermutationConf conf) noexcept;

函数功能

设置第一个Transpose操作的转置类型。

函数原型

void SetFirstTranspose(PermutationConf conf) noexcept;

约束说明

  • 该接口只支持调用一次并且只允许ShuffleLayer单独调用,或多个操作时在SetReshapeDimensions和SetSecondTranspose之前调用。
  • 默认不执行该次转置。
  • PermutationConf 的rank必须与ShuffleLayer的输入张量“rank”相同且每一个rank的取值范围为[-rank, rank - 1]。
  • conf.rank的最大值为8,在conf.order数组中,只取前“rank”个值作为Transpose操作的输入,每个值所表达的轴不允许重复。

参数说明

参数名

输入/输出

说明

conf

输入

ShuffleLayer的第一个Transpose操作配置,位于reshape之前。

返回值说明