TF代码研读-Tensor(Tensor)

橘子都吃不起！

于 2021-01-31 13:54:30 发布

阅读量656

点赞数

分类专栏： Tensorflow源码研读文章标签： tensorflow 神经网络框架 c++ 深度学习

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

版权

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

10 篇文章 ¥29.90 ¥99.00

订阅专栏

本文将详细解析TensorFlow库中Tensor类的实现，包括其定义在tensorflow/core/framework/tensor.h中的构造函数，如标量构造、拷贝与移动构造、赋值操作等。此外，还会探讨核心成员变量shape_和buf_，以及Tensor类中声明的友元类和函数，这些友元主要为了方便访问内部资源和管理内存。

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

代码研读 Tensor类

Tensor类定义在tensorflow/core/framework/tensor.h中

Tensor定义了以下几个构造函数

  //默认构造函数.初始化一个1维0元素的float张量
  Tensor();
  //另外一种构造函数Tensor(DT_FLOAT, TensorShape({}))
  Tensor(DataType type, const TensorShape& shape);  
    //指定内存分配器,构造
  Tensor(Allocator* a, DataType type, const TensorShape& shape);
    /// 注释上来说直接在op中直接调用Tensor构造函数来创建张量的方式已经弃用了,现在是使用OpKernelConstruction/OpKernelContext allocate_* 
  // 的方式来创建一个新的张量
  /// `a` must outlive the lifetime of this Tensor.
  Tensor(Allocator* a, DataType type, const TensorShape& shape,
         const AllocationAttributes& allocation_attr);
   explicit Tensor(DataType type);

针对标量(单个数字),专门定义了一个空的结构体来标记

了解本专栏