can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first
改写成这种形式 fusion_fea_0 = fusion_fea_0.data.cpu().numpy()
can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory
最新推荐文章于 2023-01-02 13:40:31 发布