Overview
Tensor is a tensor management class, which is used to store and perform operations on tensor data. This class records the shape, data format, data type, and storage location of a tensor, and provides APIs for accessing data addresses, computing data sizes, and more.
Header Files to Be Included
1 | #include <tensor.h> |
Public Member Functions
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()
Parent topic: Tensor