Shortcuts

torch.utils

rename_privateuse1_backend

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

Return 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

swap_tensors

This function swaps the content of the two Tensor objects.

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