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

torch.cuda.stream

torch.cuda.stream(stream)

paddle.device.cuda.stream_guard

paddle.device.cuda.stream_guard(stream)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | stream | stream | 指定的 CUDA stream。如果为 None,则不进行 stream 流切换。Paddle 的该 API 目前仅支持动态图模式。 |