python - 如何直接在 GPU 或另一个张量的设备上创建张量？

Question

我找到了 this关于这个的讨论，其中的代码

if std.is_cuda:
    eps = torch.FloatTensor(std.size()).cuda().normal_()
else:
    eps = torch.FloatTensor(std.size()).normal_()

变得美好

eps = std.new().normal_()

但据说那里

The new() method is deprecated.

如何直接在特定设备上创建新张量？

如何在没有丑陋的情况下像另一个张量一样在同一设备上创建新张量 if ?

Answer 1

我认为文档现在已经很清楚了。 Here描述了创建新张量的 4 种主要方法，您只需指定设备即可在 gpu 上创建它:

t1 = torch.zeros((3,3), device=torch.device('cuda'))
t2 = torch.ones_like(t1, device=torch.device('cuda'))
t3 = torch.randn((3,5), device=torch.device('cuda'))

和 this link添加了有关 torch.tensor() 的更多信息构造函数。同样，设备是要指定的参数。
如果你想使用另一个张量的设备，你可以通过 tensor.device 访问它:

t4 = torch.empty((2,2), device=t3.device)