一、问题描述 📚
在模型精度检查时，有时只需要监控特定的几个关键 Tensor 。提供一个接口，用户将调用这个接口侵入式地修改组网代码，标记需要监控的 Tensor 。这个接口与其他接口是互相独立的。
二、任务目标 🚀
该接口收到一个 Tensor 后，应当保存这个 Tensor 的信息（到某个全局变量/闭包/或者其他机制），并注册反向 hook 。在一个step 后提供某种机制，将这个 step 中记录的 Tensor 信息保存下来。
三、 TIPS
具体的做法可以参考仓库中 report 文件夹下的 report 与 tensor_hook 之间的互动。最后，还需要提供一个 dump 记录信息的接口。
一、问题描述 📚
在模型精度检查时，有时只需要监控特定的几个关键
Tensor。提供一个接口，用户将调用这个接口侵入式地修改组网代码，标记需要监控的Tensor。这个接口与其他接口是互相独立的。二、任务目标 🚀
该接口收到一个
Tensor后，应当保存这个Tensor的信息（到某个全局变量/闭包/或者其他机制），并注册反向hook。在一个step后提供某种机制，将这个step中记录的Tensor信息保存下来。三、 TIPS
具体的做法可以参考仓库中
report文件夹下的report与tensor_hook 之间的互动。最后，还需要提供一个dump记录信息的接口。