简介
Tensor是张量管理类，用于存储和操作张量数据。该类记录了张量的shape、数据格式、数据类型、存放位置，提供了数据地址访问、数据大小计算等接口。
需要包含的头文件
1 | #include <tensor.h> |
Public成员函数
Tensor() Tensor(const StorageShape &storage_shape, const StorageFormat &storage_format, const TensorPlacement placement, const ge::DataType data_type, TensorAddress addr) Tensor(const StorageShape &storage_shape, const StorageFormat &storage_format, ge::DataType data_type) Tensor(const StorageShape &storage_shape, const StorageFormat &storage_format, const TensorPlacement placement, const ge::DataType data_type, TensorAddress addr, TensorAddrManager manager) int64_t GetShapeSize() const template<class T> const T *GetData() const template<class T> auto GetData() -> T* void SetData(TensorData &&data) const void *GetAddr() const void *GetAddr() size_t GetSize() const void SetSize(const size_t size) ge::DataType GetDataType() const void SetDataType(const ge::DataType data_type) static std::unique_ptr<uint8_t[]> CreateFollowing(const int64_t shape_size, const ge::DataType dt, size_t &total_size) static std::unique_ptr<uint8_t[]> CreateFollowing(const ge::DataType dt, const size_t tensor_size, size_t &total_size) const Shape &GetStorageShape() const Shape &MutableStorageShape() const Shape &GetOriginShape() const Shape &MutableOriginShape() const StorageShape &GetShape() const StorageShape &GetShape() ge::Format GetStorageFormat() const void SetStorageFormat(const ge::Format storage_format) ge::Format GetOriginFormat() const void SetOriginFormat(const ge::Format origin_format) const StorageFormat &GetFormat() const StorageFormat &MutableFormat() ExpandDimsType GetExpandDimsType() const void SetExpandDimsType(const ExpandDimsType &expand_dims_type) TensorPlacement GetPlacement() const void SetPlacement(const TensorPlacement placement) const TensorData &GetTensorData() const TensorData &MutableTensorData()
父主题： Tensor