torch.backends.cpu 的源代码
导入 torch __all__ = ["get_cpu_capability",]
[文档]def get_cpu_capability() -> str:
返回 CPU 能力作为字符串值。
可能的值:
- "默认"
- "VSX"
Z 向量
- "无 AVX"
AVX2
AVX512
SVE256
""
返回 torch._C._get_cpu_capability()
导入 torch __all__ = ["get_cpu_capability",]
[文档]def get_cpu_capability() -> str:
返回 CPU 能力作为字符串值。
可能的值:
- "默认"
- "VSX"
Z 向量
- "无 AVX"
AVX2
AVX512
SVE256
""
返回 torch._C._get_cpu_capability()