流式上下文 ¶

class torch.cpu.StreamContext(stream)

选择给定流的上下文管理器。

注意：本类仅用于方便实现设备无关的代码