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之前。 |
返回值说明
无
父主题: class ShuffleLayer