StreamContext

class torch.cpu.StreamContext(stream)

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

N.B. This class only exists to facilitate device-agnostic code