例如，我想做一个矩阵乘法，在这样做时，我在tensorflow内部使用了tf.matmul操作。而且，我想优化tf中的矩阵乘法。但是，我无法在tf_matmul中精确地进行矩阵乘法的位置。有没有人可以帮助我做到这一点？
答案 0 :(得分：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
祝一切顺利！