小编kev*_*98x的帖子

谁能说出华莱士乘法器和达达乘法器之间的部分乘积减少方法或机制的区别？

我一直在阅读A_comparison_of_Dadda_and_Wallace_multiplier_delays.pdf

我正在使用修改后的Forecast.py来测试修剪的SqueezeNet模型

[phung@archlinux SqueezeNet-Pruning]$ python predict.py --image 3_100.jpg --model model_prunned --num_class 2
prediction in progress
Traceback (most recent call last):
File “predict.py”, line 66, in
prediction = predict_image(imagepath)
File “predict.py”, line 52, in predict_image
index = output.data.numpy().argmax()
TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
[phung@archlinux SqueezeNet-Pruning]$

我了解numpy还不支持gpu。

如何在不调用张量复制数据操作Tensor.cpu（）的情况下修改代码以摆脱此错误？