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. |