torch.cuda.default_stream

torch.cuda.default_stream(device=None)

傳回給定裝置的預設 Stream。

參數

device (torch.device 或 int, optional) – 選取的裝置。 傳回目前裝置的預設 Stream，由 current_device() 給定，如果 device 為 None (預設)。

傳回類型

Stream