SetSingleOutputShape
产品支持情况
功能说明
设置单核上结果矩阵Output的形状。
Conv3D高阶API目前支持M合轴模式的输出方式。在M合轴模式下,Conv3D API内部将Wout和Hout视为同一轴处理,输出时先沿Wout方向输出,完成一整行Wout输出后,再进行下一行的Wout输出。
图 1M合轴模式示意图

函数原型
__aicore__ inline void SetSingleOutputShape(uint64_t singleCo, uint64_t singleDo, uint64_t singleM)参数说明
返回值说明
无
约束说明
本接口当前仅支持设置Output的C维度、D维度和M维度(即H轴、W轴合并后的维度),不支持设置原始Output的大小。
调用示例
conv3dApi.SetSingleOutputShape(singleCoreCout, singleCoreDout, singleCoreM);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考