在指定三维输出shape信息(outputSize)的情况下,完成张量self的3D自适应平均池化计算。aclnnAdaptiveAvgPool3d与aclnnAvgPool3d不同的是,aclnnAdaptiveAvgPool3d只需要指定输出的大小,就可以自动推导出kernel的大小与对应的步长。
每个算子分为,必须先调用“aclnnAdaptiveAvgPool3dGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnAdaptiveAvgPool3d”接口执行计算。
[object Object]
[object Object]
参数说明:
[object Object]- [object Object]Atlas 推理系列产品[object Object]:参数
[object Object]、[object Object]的数据类型不支持BFLOAT16。
- [object Object]Atlas 推理系列产品[object Object]:参数
返回值
aclnnStatus:返回状态码,具体参见。 第一段接口完成入参校验,出现以下场景时报错:
[object Object][object Object]- [object Object]Atlas 推理系列产品[object Object]:参数
[object Object]、[object Object]维度不支持第一维度不大于0。
- [object Object]Atlas 推理系列产品[object Object]:参数
确定性计算:
- aclnnAdaptiveAvgPool3d默认确定性实现。
Shape描述:
- self.shape = (N, C, Din, Hin, Win)或者(C, Din, Hin, Win)
- outputSize = [Dout, Hout, Wout]
- out.shape = (N, C, Dout, Hout, Wout)或者(C, Dout, Hout, Wout)
[object Object]