下载
中文
注册

GetStorageShape

函数功能

获取运行时Tensor的Storageshape,此shape对象为只读。Storgeshape和Originshape的区别如下:Originshape是Tensor最初创建时的形状,StorageShape是保存Tensor数据的底层存储的形状。运行时为了适配底层硬件,Tensor的StorageShape和其Originshape可能会有所不同。

函数原型

const Shape &GetStorageShape() const

参数说明

无。

返回值说明

只读的运行时shape引用。

约束说明

无。

调用示例

StorageShape sh({1, 2, 3}, {2, 1, 3});
Tensor t = {sh, {}, {}, ge::DT_FLOAT, nullptr};
auto shape = t.GetStorageShape(); // 2,1,3