MemPoolContext

class torch.cuda.MemPoolContext(*args, **kwargs)

MemPoolContext 保留目前作用中的集區，並儲存先前的集區。刪除時，會使先前的集區變為作用中。

參數

• pool (torch.cuda.MemPool) – 要使其處於作用中的 MemPool 物件，因此

• pool. (分配路由至此) –

static active_pool()

傳回作用中的 MemPool

回傳類型

Optional[_MemPool]