下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词