设置Stride信息。
1 | void SetStride(int64_t strideD, int64_t strideH, int64_t strideW) |
参数名 |
输入/输出 |
描述 |
|---|---|---|
strideD |
输入 |
卷积正向过程中Depth方向Stride的大小。 |
strideH |
输入 |
卷积正向过程中Height方向Stride的大小。 |
strideW |
输入 |
卷积正向过程中Width方向Stride的大小。 |
无
无
1 2 3 | auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3DBpInputTiling conv3DBpDxTiling(*ascendcPlatform); conv3DBpDxTiling.SetStride(strideD, strideH, strideW); |