跳到主要内容

SetDim

函数功能​

设置dim值。

函数原型​

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