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

特性范围

数据类型

表1 支持的参数数据类型

src数据类型

dst数据类型

f32

f32

f16

f16

bf16

bf16

s32

s32

s8

s8

u8

u8

数据排布

支持1D~5D Tensor,支持以下数据排布,输入和输出的排布必须相同。

Tensor维度

src0/src1数据排布

dst数据排布

1D Tensor

a

a

2D Tensor

ab,ba

ab,ba

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

参数约束