[参数不一致]torch.cuda.Stream¶
torch.cuda.Stream¶
torch.cuda.Stream(device=None, priority=0, **kwargs)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
device | device | 希望分配 stream 的设备。 |
priority | priority | stream 的优先级,PyTorch 取值范围为-1、0,Paddle 的取值范围为 1、2,Paddle 需要进行转写。 |