[参数完全一致]torch.cuda.current_stream¶
torch.cuda.current_stream¶
torch.cuda.current_stream(device=None)
参数映射¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | device | device | 表示希望获取 stream 的设备或者设备 ID。如果为 None,则为当前的设备。默认值为 None。 |