TensorFlow源码分析（2）：ConstantOp类的构造函数

最新推荐文章于 2021-12-29 22:38:50 发布

汪星人来地球

最新推荐文章于 2021-12-29 22:38:50 发布

阅读量612

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/hedan2013/article/details/103373901

版权

本文深入探讨了TensorFlow中ConstantOp类的构造函数，主要逻辑包括：调用父类OpKernel的初始化方法，创建并初始化Tensor对象，将TensorProto转换为Tensor，以及进行类型检查确保一致。同时，解释了OP_REQUIRES_OK和OP_REQUIRES宏的作用，以及它们如何利用编译优化提升性能。

摘要由CSDN通过智能技术生成

在文章（1）中，我们说到TensorFlow的操作类中，最重要的两个函数是构造函数和Compute函数。这篇文章，先介绍ConstantOp类的构造函数。它定义在tensorflow/core/kernels/constant_op.cc文件中。

ConstantOp::ConstantOp(OpKernelConstruction* ctx)
    : OpKernel(ctx, StripTensorDataFromNodeDef(ctx)),
      tensor_(ctx->output_type(0)) {
  const TensorProto* proto = nullptr;
  OP_REQUIRES_OK(ctx, ctx->GetAttr("value", &proto));
  OP_REQUIRES_OK(ctx, ctx->device()->MakeTensorFromProto(
                          *proto, AllocatorAttributes(), &tensor_));
  OP_REQUIRES(
      ctx, ctx->output_type(0) == tensor_.dtype(),
      errors::InvalidArgument("Type mismatch between value (",
                              DataTypeString(tensor_.dtype()), ") and dtype (",
                              DataTypeString(ctx->output_type(0)), ")"));
}
</