torch.utils¶
rename_privateuse1_backend |
将privateuse1后端设备重命名为更方便在PyTorch API中使用的设备名称。 |
generate_methods_for_privateuse1_backend |
在重命名 privateuse1 后端后,自动生成属性和方法以供自定义后端使用。 |
get_cpp_backtrace |
返回包含当前线程的C++堆栈跟踪的字符串。 |
set_module |
为给定对象的python对象设置模块属性,以获得更好的打印效果 |
swap_tensors |
此函数交换两个 Tensor 对象的内容。 |