StreamContext

class torch.cpu.StreamContext(stream)

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

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