通过字符串: >>> torch.device('cuda:0') device(type='cuda', ... 需要注意的是,torch.tensor 总是会复制data, 如果你想避免复制,可以使torch.
確定! 回上一頁