Tensor API¶

  // 构造 DenseTensor 并分配内存
  // 参数：a - Allocator 指针类型
  //      meta - DenseTensorMeta 对象
  // 返回：None
  DenseTensor(Allocator* a, const DenseTensorMeta& meta);

  // 构造 DenseTensor 并分配内存
  // 参数：a - Allocator 指针类型
  //      meta - DenseTensorMeta 移动对象
  // 返回：None
  DenseTensor(Allocator* a, DenseTensorMeta&& meta);

  // 构造 DenseTensor 并分配内存
  // 参数：holder - Allocation 共享指针类型
  //      meta - DenseTensorMeta 移动对象
  // 返回：None
  DenseTensor(const std::shared_ptr<phi::Allocation>& holder,
              const DenseTensorMeta& meta);

  // 移动构造函数
  // 参数：other - DenseTensor 移动对象
  // 返回：None
  DenseTensor(DenseTensor&& other) = default;

  // 拷贝构造函数
  // 参数：other - DenseTensor 对象
  // 返回：None
  DenseTensor(const DenseTensor& other);

  // 赋值操作
  // 参数：other - DenseTensor 对象
  // 返回：DenseTensor 对象
  DenseTensor& operator=(const DenseTensor& other);

  // 移动赋值操作
  // 参数：other - DenseTensor 对象
  // 返回：DenseTensor 对象
  DenseTensor& operator=(DenseTensor&& other);

  // 无参构造函数
  DenseTensor();

  // 析构函数
  virtual ~DenseTensor() = default;

  // 获取类名，静态函数
  // 参数：None
  // 返回：字符串指针
  static const char* name();

  // 获取 Tensor 中元素数量
  // 参数：None
  // 返回：int64_t 类型变量
  int64_t numel() const override;

  // 获取 Tensor 的 dims 信息
  // 参数：None
  // 返回：DDim 对象
  const DDim& dims() const noexcept override;

  // 获取 Tensor 的 lod 信息
  // 参数：None
  // 返回：LoD 对象
  const LoD& lod() const noexcept;

  // 获取 Tensor 的数据类型信息
  // 参数：None
  // 返回：DataType 类型变量
  DataType dtype() const noexcept override;

  // 获取 Tensor 的内存布局信息
  // 参数：None
  // 返回：DataLayout 类型变量
  DataLayout layout() const noexcept override;

  // 获取 Tensor 的 Place 信息
  // 参数：None
  // 返回：Place 类型变量
  const Place& place() const override;

  // 获取 Tensor 的 meta 信息
  // 参数：None
  // 返回：DenseTensorMeta 对象
  const DenseTensorMeta& meta() const noexcept;

  // 设置 Tensor 的 meta 信息
  // 参数：meta - DenseTensorMeta 移动对象
  // 返回：None
  void set_meta(DenseTensorMeta&& meta);

  // 设置 Tensor 的 meta 信息
  // 参数：meta - DenseTensorMeta 对象
  // 返回：None
  void set_meta(const DenseTensorMeta& meta);

  // 检查 Tensor 的 meta 信息是否有效
  // 参数：None
  // 返回：bool 类型变量
  bool valid() const noexcept override;

  // 检查 Tensor 的是否被初始化
  // 参数：None
  // 返回：bool 类型变量
  bool initialized() const override;

  // 为 Tensor 分配内存
  // 参数：allocator - Allocator 类型指针
  //      dtype - DataType 变量
  //      requested_size - size_t 类型变量
  // 返回：void*类型指针
  void* AllocateFrom(Allocator* allocator,
                     DataType dtype,
                     size_t requested_size = 0) override;

  // 检查是否与其它 Tensor 共享内存
  // 参数：b - DenseTensor 对象
  // 返回：bool 类型变量
  bool IsSharedWith(const DenseTensor& b) const;

  // 修改 Tensor 的 Dims 信息并分配内存
  // 参数：dims - DDim 对象
  // 返回：None
  void ResizeAndAllocate(const DDim& dims);

  // 修改 Tensor 的 Dims 信息
  // 参数：dims - DDim 对象
  // 返回：DenseTensor 对象
  DenseTensor& Resize(const DDim& dims);

  // 重置 Tensor 的 LoD 信息
  // 参数：lod - LoD 对象
  // 返回：None
  void ResetLoD(const LoD& lod);

  // 获取 Tensor 的内存大小
  // 参数：None
  // 返回：size_t 类型变量
  size_t capacity() const;

  // 获取 Tensor 的不可修改数据指针
  // 模板参数：T - 数据类型
  // 参数：None
  // 返回：不可修改的 T 类型数据指针
  template <typename T>
  const T* data() const;

  // 获取 Tensor 的不可修改数据指针
  // 参数：None
  // 返回：不可修改的 void 类型数据指针
  const void* data() const;

  // 获取 Tensor 的可修改内存数据指针
  // 模板参数：T - 数据类型
  // 参数：None
  // 返回：可修改的 T 类型数据指针
  template <typename T>
  T* data();

  // 获取 Tensor 的可修改内存数据指针
  // 参数：None
  // 返回：可修改的 void 类型数据指针
  void* data();