Hook函数机制:不改变主体实现额外功能,像一个挂件挂钩,hook
功能 注册一个反向传播hook函数 Hook函数仅一个输入参数为张量的梯度
input 当前网络层输入数据 output 当前网络层输出数据
input 当湔网络层输入数据
对网络的最后一个特征图进行加权求和 得到注意力机制
CAM改进版,利用梯度作为特征图权重