torch.cuda.graph_pool_handle

torch.cuda.graph_pool_handle()[源代码]

返回一个代表图内存池ID的不透明令牌。

参见图内存管理。

警告

此 API 处于 beta 阶段，未来版本可能有所更改。