SetDim

Description

Sets the dimension value of an axis.

Prototype

void SetDim(size_t idx, const int64_t dim_value)

Parameters

Parameter

Input/Output

Description

idx

Input

Index of dim. You must ensure that the index is valid.

dim_value

Input

Dimension value set for the idx axis.

Returns

None

Restrictions

You must ensure that the index is valid.

Example

Shape shape0({3, 256, 256});
shape0.SetDim(0U, 1); // 1,256,256