[参数不一致]torch.cuda.Stream¶

torch.cuda.Stream ¶

          torch.cuda.Stream(device=None, priority=0, **kwargs)

         

paddle.device.cuda.Stream ¶

          paddle.device.cuda.Stream(device=None, priority=None)

         

两者功能一致，参数不一致，具体如下：

参数映射¶

PyTorch	PaddlePaddle	备注
device	device	希望分配 stream 的设备。
priority	priority	stream 的优先级，PyTorch 取值范围为-1、0，Paddle 的取值范围为 1、2，需要转写。

转写示例¶

priority: stream 的优先级¶

           # Pytorch 写法
high_priority = -1
default_priority = 0
y = torch.cuda.Stream(priority=default_priority)

# Paddle 写法
high_priority = 1
default_priority = 2
y = paddle.device.cuda.Stream(priority=default_priority)

          

[参数不一致]torch.cuda.Stream¶

torch.cuda.Stream¶

paddle.device.cuda.Stream¶

参数映射¶

转写示例¶

priority: stream 的优先级¶

torch.cuda.Stream ¶

paddle.device.cuda.Stream ¶