SetInputShape

Function

Sets the shape of the feature matrix input, including Batch, Channel, Depth, Height, and Width. When constructing the Conv3DTranspose operator, this API is not applicable. Do not use it.

Prototype

1

bool SetInputShape(int64_t n, int64_t c, int64_t d, int64_t h, int64_t w)

Parameters

Returns

true indicates that the setting is successful, and false indicates that the setting fails.

Restrictions

None

Example

1
2
3

auto ascendcPlatform = platform_ascendc::PlatformAscendCManager::GetInstance();
ConvBackpropApi::Conv3DBpInputTiling conv3DBpDxTiling(*ascendcPlatform);
conv3DBpDxTiling.SetInputShape(n, c, d, h, w);