鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

特性范围

数据类型

传播方向

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