简介
该类描述了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
父主题: StorageShape