torch.cuda.graph_pool_handle

torch.cuda.graph_pool_handle()

返回一个不透明的标记，代表图内存池的 ID。

参见 图内存管理。

警告

此 API 为测试版，未来版本中可能会发生变化。