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')