Overview
This class describes shapes of a tensor, including origin_shape and storage_shape. origin_shape indicates the mathematical description of a tensor shape, while storage_shape indicates the actual shape format of origin_shape at runtime.
Header Files to Be Included
1 | #include <storage_shape.h> |
Public Member Functions
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
Parent topic: StorageShape