torch.xpu.memory.memory_reserved

torch.xpu.memory.memory_reserved(device=None)

返回给定设备上缓存分配器管理的当前 GPU 内存（以字节为单位）。

参数

device (torch.device or int or str, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None (default).

返回类型

int