GetShape
函数功能
获取Tensor的shape,包含运行时和原始shape。
函数原型
const StorageShape &GetShape() const
StorageShape &GetShape()
参数说明
无。
返回值说明
- const StorageShape &GetShape() const:返回只读的shape引用。
- StorageShape &GetShape():返回shape引用。
关于StorageShape类型的定义,请参见StorageShape结构体。
约束说明
无。
调用示例
StorageShape sh({1, 2, 3}, {2, 1, 3}); Tensor t = {sh, {}, {}, ge::DT_FLOAT, nullptr}; auto shape = t.GetShape(); // sh
父主题: Tensor类