我要评分

文档获取效率

文档正确性

内容完整性

文档易理解

在线提单

论坛求助

特性范围

数据类型

传播方向	src数据类型	dst数据类型	alg类型
forward/backward	f32	f32	max, avg_p, avg_np
forward/backward	f16	f16	max, avg_p, avg_np
forward/backward	bf16	bf16	max, avg_p, avg_np
forward	s8	s8	max, avg_p, avg_np
forward	u8	u8	max, avg_p, avg_np
forward	s32	s32	max, avg_p, avg_np

数据排布

支持1D~5D Tensor，支持以下数据排布。

Tensor维度	src数据排布	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

参数约束

无

父主题： Pool