Python 如何在TensorFlow中监视cblas调用？

我正在从事一个项目，我有一个.so（共享对象），其中包含一组处理矩阵加法和矩阵乘法的函数。我在运行一个使用TensorFlow的python脚本时使用了这个.so文件。 简而言之，我使用的不是TensorFlow中内置的矩阵乘法函数，而是具有矩阵乘法函数的.so文件（启用TensorFlow mkl使用libmklml_intel.so文件实现矩阵乘法函数）。是mkl采用矩阵乘法的格式

现在，眼前的问题是我需要监控三件事：

相乘矩阵的维数是多少

这个函数被调用了多少次

脚本在此函数中花费的%if时间运行时间是多少

我可以想出两种方法

跟踪从TensorFlow框架进行这些函数调用的位置，然后进行更改以获得所需的信息。但我觉得这种方法需要在多个地方进行修改，因为调用cblas的函数不只是一个。此外，我还必须勉强读完TensorFlow源代码，这是一项艰巨的任务。它还可能要求再次从源头上建立TF

但是，如果我们能够以某种方式监视.so文件与TF的交互，并且可能以某种方式注册调用时发送的所有数据，可能使用一个聪明的shell脚本，或者使用一个我不知道的工具？我最好的选择是什么？这有什么帮助吗

提前谢谢 这里是新手