torch.accelerator.current_stream#
- torch.accelerator.current_stream(device=None, /)[源码]#
返回指定设备的当前选定流。
- 参数
device (
torch.device, str, int, optional) – 指定的设备，该设备必须与当前的 accelerator 设备类型相匹配。如果未给出，则默认使用torch.accelerator.current_device_index()。- 返回
指定设备的当前选定流。
- 返回类型
