
SetWeightShape功能说明设置权重矩阵Weight的形状。函数原型void SetWeightShape(int64_t cout, int64_t cin, int64_t d, int64_t h, int64_t w)参数说明表 1参数说明参数名输入/输出描述cout输入设置GradOutput的Channel值。cin输入设置Input的Channel值。d输入设置Weight的Depth值。h输入设置Weight的Height值。w输入设置Weight的Width值。返回值说明无约束说明无调用示例optiling::Conv3DBackpropFilterTilingData tilingData; auto ascendcPlatform platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetWeightShape(cout, cin, d, h, w);创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考