跳到主要内容

GetStorageShape

函数功能​

获取运行时Tensor的StorageShape,此shape对象为只读。StorageShape和GetOriginShape的区别如下。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