data1 = torch.randn(1, 3, device=d2l.try_gpu(0))
data2 = torch.randn(1, 3, device=d2l.try_gpu(1))
# 打印原始张量的内容和设备
print("data[0]:", data1)
print("data[0] device:", data1.device)
print("data[1]:", data2)
print("data[1] device:", data2.device)
# 使用 .to() 方法并打印
data1_to_device0 = data2.to(data1.device)
print("data[1] after .to(data[0].device):", data1_to_device0)
print("device of the new tensor:", data1_to_device0.device)
代码如上,在运行过程中data1_to_device0 = data2.to(data1.device)这一句在to其他device中居然把数值给修改了,这是为什么啊哪里出错了,改成了device0中的数据了。运行结果如下:
