评价此页

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 对象的模块属性,以便更好地打印

swap_tensors

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