请问下怎么在.cu文件中怎么把gpu_data转换为cpu_data,并且打印出来?

我想把计算loss的两个值里面的data打印出来,因为采用的是GPU,只需考虑Forward_gpu()方法中,把gpu_data转换为cpu_data,然后打印出来?
已邀请:

陶潜水

赞同来自:

可以试下直接调用cpu_data的数据进行打印,如果不行就用cudamemcpy把GPU数据拷贝到CPU

要回复问题请先登录注册