Python PyTorch Generator用法及代码示例

本文简要介绍python语言中 torch.Generator 的用法。

用法:

class torch.Generator(device='cpu') → Generator

参数：

device(torch.device， 可选的) -生成器所需的设备。

返回：

一个 torch.Generator 对象。

返回类型：

生成器

创建并返回一个生成器对象，该对象管理生成伪随机数的算法的状态。在许多就地随机抽样函数中用作关键字参数。

例子：

>>> g_cpu = torch.Generator()
>>> g_cuda = torch.Generator(device='cuda')