torch.cpu.stream

torch.cpu.stream(stream)

圍繞 StreamContext 的包裝器，用於選擇指定的流。

注意：此函式僅用於方便裝置無關的程式碼

返回型別

AbstractContextManager