SetViewShape
函数功能
设置aclTensor的ViewShape。ViewShape表示aclTensor的逻辑shape,可由GetViewShape获取。
函数原型
void SetViewShape(const op::Shape &shape)
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
shape |
输入 |
数据类型为op::Shape(即gert::Shape),记录了一组shape信息,例如一个三维shape:[10, 20, 30]。 |
返回值说明
无
约束说明
无
调用示例
// 将input的ViewShape设置为[1, 2, 3, 4, 5]
void Func(aclTensor *input) {
gert::Shape newShape;
for (int64_t i = 1; i <= 5; i++) {
newShape.AppendDim(i);
}
input->SetViewShape(newShape);
}
父主题: common_types