TF代码研读-Tensor(TensorBuffer,RefCounted)

橘子都吃不起！

于 2021-01-31 13:41:05 发布

阅读量696

点赞数

分类专栏： Tensorflow源码研读文章标签：设计模式多态 c++ tensorflow 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_42754237/article/details/113469812

版权

Tensorflow源码研读专栏收录该内容

10 篇文章 ¥29.90 ¥99.00

订阅专栏

本文深入探讨了TensorFlow中的Tensor类，其内部核心成员变量TensorShape和TensorBuffer。TensorBuffer继承自RefCounted类，用于管理内存，并通过引用计数实现buffer的生命周期管理。TensorBuffer定义了纯虚函数，强制通过子类实例化以实现多态。在Tensor类中，通过HostScalarTensorBufferBase和模板结构体ValueAndTensorBuffer，创建了具体的TensorBuffer子类并实现了所需功能，确保了灵活性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Tensor类中包含了两个核心成员变量 TensorShape shape_;

TensorBuffer* buf_;其中TensorBuffer类型变量定义在tensor.h文件中,继承RefCounted类.RefCounted类是一个引用计数器类.

class RefCounted {
 public:
  // Initial reference count is one.
  RefCounted();//引用计数器类,引用计数初始化为1
  // Increments reference count by one.
  void Ref() const;   //引用计数加一
  // Decrements reference count by one.  If the count remains
  // positive, returns false.  When the count reaches zero, returns
  // true and deletes this, in which case the caller must not access
  // the object afterward.
  bool Unref() const;//引用计数减一,当计数为负值时返回false
  bool RefCountIsOne() const;//判断引用计数是否为1

 protected:
  // Make destructor protected so that RefCounted objects cannot
  // be instantiated directly. Only subclas

了解本专栏