torch.accelerator.current_stream¶
- torch.accelerator.current_stream(device=None, /)[源][源]¶
返回給定裝置的當前選中流。
- 引數
device (
torch.device, str, int, 可選) – 一個給定的裝置，必須與當前的 accelerator 裝置型別匹配。如果未給出，預設使用torch.accelerator.current_device_index()。- 返回
給定裝置的當前選中流。
- 返回型別
