torchsnooper输出torch中的变量类型/维度

1. 输出函数中的内容

import torchsnooper

@torchsnooper.snoop()
def func(x,y):
	...
	return 

在函数前加@torchsnooper.snoop()

2. 输出语句中的内容

import torchsnooper

with torchsnooper.snoop():
	语句1
	语句2
	···
	

在语句前使用with torchsnooper.snoop():
下图是一张结果图。