ToShape

函数功能

将指定内存维度数据转为op::Shape（即gert::Shape）对象。

函数原型

• 用给定的dims维度数组构造op::Shape。

  void ToShape(const int64_t *dims, uint64_t dimNum, Shape &shape)

• 用给定的op::ShapeVector维度数组构造op::Shape。

  void ToShape(const FVector<int64_t, 25> shapeVector, Shape &shape)

参数说明

返回值说明

无

约束说明

无

调用示例

1
2
3
4
5
6

// 生成一个shape信息为[1, 2, 3, 4, 5]的Shape对象。
void Func() {
    int64_t myArray[5] = {1, 2, 3, 4, 5};
    Shape shape;
    ToShape(myArray, 5, shape);
}