torch.mtia.default_stream

torch.mtia.default_stream(device=None)

傳回給定裝置的預設 Stream。

參數

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

傳回類型

Stream