Tensor.copy_ 和其他具有類似複製功能的方法(如 ~torch. ... index=0) with torch.cuda.device(1): # allocates a tensor on GPU 1 a ...
確定! 回上一頁