Shortcuts

Source code for torch.backends.cpu

import torch

__all__ = [
    "get_cpu_capability",
]


[docs]def get_cpu_capability() -> str: r"""Return cpu capability as a string value. Possible values: - "DEFAULT" - "VSX" - "Z VECTOR" - "NO AVX" - "AVX2" - "AVX512" """ return torch._C._get_cpu_capability()

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources