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

torch.cuda.current_stream ¶

torch.cuda.current_stream(device=None)

paddle.device.cuda.current_stream(device=None)

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

PyTorch	PaddlePaddle	备注
device	device	表示希望获取 stream 的设备或者设备 ID。如果为 None，则为当前的设备。默认值为 None。