pytorch 同步gpu
import torch a = torch.tensor([[1, 2, 3],[4, 5, 6]]) b = torch.tensor([[2, 2, 2], [3, 3, 3], [3, 3, 3], [3, 3, 3]]) a=a.to(b.device)
pytorch 同步gpu
import torch a = torch.tensor([[1, 2, 3],[4, 5, 6]]) b = torch.tensor([[2, 2, 2], [3, 3, 3], [3, 3, 3], [3, 3, 3]]) a=a.to(b.device)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
