torch.mps.device_count()

torch.mps.device_count()[源代码]

返回可用的MPS设备的数量。

返回类型

int

本页目录