PyTorch代码调试利器:自动print每行代码的Tensor信息

安装完了以后,只需要用@torchsnooper.snoop()装饰一下要调试的函数,这个函数在执行的时候,就会自动print出来每一行的执行结果的tensor的形状、数据类型、设备、是否需要梯度的信息。安装完了以后,下面就用两个例子来说明一下怎么使用。例子1比如说我们写了一个非常简单的函数:defmyfunc(mask,x):y...