昇腾社区首页
中文
注册

简介

该类描述了tensor的shape,包含两个信息:origin_shape以及storage_shape。其中,origin_shape表示tensor的shape数学描述,storage_shape表示origin_shape实际运行时的shape格式。

需要包含的头文件

1
#include <storage_shape.h>

Public成员函数

StorageShape()
StorageShape(const std::initializer_list<int64_t> &origin_shape, const std::initializer_list<int64_t> &storage_shape)
const Shape &GetOriginShape() const
const Shape &GetStorageShape() const
Shape &MutableOriginShape()
Shape &MutableStorageShape()
bool operator==(const StorageShape &other) const
bool operator!=(const StorageShape &other) const