设置某一个轴的维度值。
void SetDim(size_t idx, const int64_t dim_value)
参数
输入/输出
说明
idx
输入
dim的index,调用者需要保证index合法。
dim_value
对idx轴设置的维度值。
无。
调用者需要保证index合法。
Shape shape0({3, 256, 256}); shape0.SetDim(0U, 1); // 1,256,256