TensorFlow中的核心运算符MatMul的实现过程系列介绍之一
发布时间
阅读量:
阅读量
本文旨在以TensorFlow中MatMul这一具备典型性且相对简单的操作为例,阐述TensorFlow中图结构节点的实现方式。笔者认为,TensorFlow中的操作(ops)构成了整个框架的核心部分,若能深入理解这部分内容,将有助于形成对TensorFlow更为深刻的认识。
在研读相关代码之前,建议先查阅官方文档中关于如何添加新操作的说明,这对后续理解具有重要参考价值:
中文翻译:http://www.tensorfly.cn/tfdoc/how_tos/adding_an_op.html
英文原文:http://www.tensorflow.org/how_tos/adding_an_op/index.html#adding-a-new-op
在对新增操作的方法有了初步了解之后,我们将通过一个实际可用的案例进行分析:MatMul
一个操作通常由两个计算节点构成,分别为正向传播计算节点和反向传播梯度计算节点。
正向传播计算节点对应的源代码位于:core/kernels/matmul_op.cc这一文件中。
/* Copyright 2015 The TensorFlow Authors. All Rights Reserved.
3. Licensed under the Apache License, Version 2.0
全部评论 (0)
还没有任何评论哟~
