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

特性范围

数据类型

表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

参数约束