\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 。