关于使用 hook 获取神经网络内部features时遇到的问题和解决方案的记录【代码】

关于使用 hook 获取神经网络内部features时遇到的问题和解决方案的记录 一、错误代码： feat = [] def hook(module, inputs, outputs):feat.append(inputs.clone().to(device))二、报错： AttributeError: 'tuple' object has no attribute 'clone' 原因分析：inputs是一个由多个Tensor类型数据作为元素组成的Tuple数据。 三、解决方案思考： 由于后续操作都需要使用Tensor数据， 而直接使用torch.FloatTensor(Tuple)将Tuple转换为...