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