使用torch随机生成一组张量并保存在GPU中

使用 torch 随机生成一组张量并保存在 GPU 中，可以使用以下方法：

import torch

# 生成随机张量
tensor = torch.rand(3, 3)

# 将张量保存在 GPU 中
tensor = tensor.to('cuda')

在这里，我们生成了一个 3x3 的随机张量，并将它转换为在 GPU 上运行的张量。需要注意的是，在使用这段代码之前，你需要确保你的计算机已经安装了 GPU 并且已经正确配置了 PyTorch 以使用 GPU。