Shortcuts

torch.utils

rename_privateuse1_backend

This API should be use to rename the privateuse1 backend device to make it more convenient to use as a device name within PyTorch APIs.

generate_methods_for_privateuse1_backend

Automatically generate attributes and methods for the custom backend after rename privateuse1 backend.

get_cpp_backtrace

Returns a string containing the C++ stack trace of the current thread.

set_module

Set the module attribute on a python object for a given object for nicer printing

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