SetInputShape
功能说明
设置特征矩阵Input的形状：Batch、Channel、Depth、Height、Width。
函数原型
1 | void SetInputShape(int64_t n, int64_t c, int64_t d, int64_t h, int64_t w) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
n |
输入 |
输入Input的Batch值。 |
c |
输入 |
输入Input的Channel值。 |
d |
输入 |
输入Input的Depth值。 |
h |
输入 |
输入Input的Height值。 |
w |
输入 |
输入Input的Width值。 |
返回值说明
无
约束说明
无
调用示例
1 2 3 | auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance(); ConvBackpropApi::Conv3dBpFilterTiling conv3dBpDwTiling(*ascendcPlatform); conv3dBpDwTiling.SetInputShape(n, c, d, h, w); |