hook和class区别

Hook 和 Class 是 PyTorch 中常用的两种机制，主要的区别在于它们的输入和输出。Hook 在一个网络结构的某个 Tensor 中植入一个函数，该函数可以截取该 Tensor 的梯度，而不对原有网络结构造成任何改变。而 Class 则是一种新的 PyTorch 数据类型，它允许用户自定义数据类型，并为这些类型定义行为，比如定义 forward、backward 等方法。