[组合替代实现]torch.Generator¶

torch.Generator ¶

          torch.Generator(device='cpu')

         

Paddle 无此 API，需要组合实现。

转写示例¶

          # PyTorch 写法
torch.Generator()

# Paddle 写法
paddle.framework.core.default_cpu_generator()

          # PyTorch 写法
torch.Generator(device="cuda")

# Paddle 写法
device = paddle.device.get_device()
paddle.framework.core.default_cuda_generator(int(device[-1]))

         