设置Dilation信息,即卷积核Depth/Height/Width方向的扩张大小。
1 | void SetDilation(int64_t dilationD, int64_t dilationH, int64_t dilationW) |
参数名 |
输入/输出 |
描述 |
|---|---|---|
dilationD |
输入 |
卷积核Weight Depth方向扩张大小。 |
dilationH |
输入 |
卷积核Weight Height方向扩张大小。 |
dilationW |
输入 |
卷积核Weight Width方向扩张大小。 |
无
无
1 2 3 | auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3DBpInputTiling conv3DBpDxTiling(*ascendcPlatform); conv3DBpDxTiling.SetDilation(dilationD, dilationH, dilationW); |