torch.utils

rename_privateuse1_backend

将privateuse1后端设备重命名,使其在PyTorch API中更便于用作设备名称。

generate_methods_for_privateuse1_backend

在将后端重命名为 privateuse1 之后,自动为自定义后端生成属性和方法。

get_cpp_backtrace

返回当前线程的C++堆栈跟踪的字符串。

set_module

为给定对象在Python对象上设置模块属性,以实现更美观的打印效果

swap_tensors

此函数用于交换两个张量对象的内容。

本页目录