[参数完全一致]torch.cuda.set_stream

torch.cuda.set_stream

torch.cuda.set_stream(stream)

paddle.device.set_stream

paddle.device.set_stream(stream=None)

功能一致,参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
stream stream 希望设置的 stream。