特性范围
数据类型
传播方向 |
src / dst数据类型 |
|---|---|
forward/backward |
f32, bf16, f16 |
数据排布
支持3D~5D Tensor，支持以下数据排布，输入和输出排布相同。
Tensor维度 |
src数据排布 |
dst数据排布 |
|---|---|---|
3D Tensor |
abc,acb,bac,bca,cab,cba |
abc,acb,bac,bca,cab,cba |
4D Tensor |
abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab |
abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab |
5D Tensor |
abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab |
abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab |
参数约束
无