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

torch.cuda.stream ¶

torch.cuda.stream(stream)

paddle.device.stream_guard(stream)

功能一致，参数完全一致，具体如下：

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