\u200E

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

torch.cuda.StreamContext ¶

          torch.cuda.StreamContext(stream)

         

paddle.device.stream_guard ¶

          paddle.device.stream_guard(stream)

         

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

参数映射¶

PyTorch	PaddlePaddle	备注
stream	stream	指定的 CUDA stream 。