SetDimSizes

函数功能

设置Tensor的shape的值，即Tensor的数据维度大小，用std::vector<int64_t>表征每一个维度的具体大小。

函数原型

void SetDimSizes(const std::vector<int64_t> &dims)

参数说明

返回值说明

无。

约束说明

无。

调用示例

vector<int64_t> dims = { 3, 1, 3, 5 };
shape->SetDimSizes(dims);