设置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