原文:张量与数组运算报错(Use Tensor.cpu() to copy the tensor to host memory first;RuntimeError: Expected all tensors to be on the same device)

结论: .张量与数组运算,张量必须在cpu上,产生结果为cpu上的张量,可继续与数组运算 张量必须在gpu上 .张量与张量运算,cpu上的张量与gpu上的张量是无法运行的,必须在相同的gpu上或cpu上,猜想不同型号的gpu因该也不行。 一.张量与数组运算,前提张量必须在cpu上 如果张量在gpu上,则会报错 我认为数组本身在cpu上,因此 个操作在cpu上,就可以默认运行 ,运算结果将会转到cp ...

2021-03-16 01:04 0 6880 推荐指数:

查看详情

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM