我在哪里可以确切地找到Tensorflow如何进行矩阵乘法？

Question

例如，我想做一个矩阵乘法，在这样做时，我在tensorflow内部使用了tf.matmul操作。而且，我想优化tf中的矩阵乘法。但是，我无法在tf_matmul中精确地进行矩阵乘法的位置。有没有人可以帮助我做到这一点？

Answer 1

我们需要进行一些代码跟踪，以找出正在调用的内容和正在发生的情况

1）tensorflow.python.ops.math_ops通过tf.matmul呼叫

2）tf.matmul返回sparse_matmul（调用gen_math_ops.sparse_matmul）或gen_math_ops.batch_mat_mul

3）gen_math_ops脚本是自动生成的，但是基础代码是math_ops.cc

祝一切顺利！