torch.xpu.get_device_name

torch.xpu.get_device_name(device=None)

获取设备的名称。

参数

device (torch.device or int or str, optional) – 需要返回名称的设备。如果此参数为负整数，则此函数为空操作。如果 device 为 None (默认)，则使用 current_device() 指定的当前设备。

返回值

设备的名称

返回类型

str