评价此页

torch.utils#

创建于:2023年7月21日 | 最后更新于:2025年6月6日

rename_privateuse1_backend

重命名 privateuse1 后端设备,以便在 PyTorch API 中更方便地用作设备名称。

generate_methods_for_privateuse1_backend

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

get_cpp_backtrace

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

set_module

为给定对象设置 python 对象的 module 属性,以便更美观地打印。

swap_tensors

此函数交换两个 Tensor 对象的内容。