SetInputShape

Function Usage

Sets the shape of the feature matrix input, including Batch, Channel, Depth, Height, and Width. When constructing the Conv3DTranspose operator, this API has no actual meaning. 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

Examples

1
2
3

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