Python PyTorch get_testing_overrides用法及代碼示例

本文簡要介紹python語言中 torch.overrides.get_testing_overrides 的用法。

用法:

torch.overrides.get_testing_overrides()

返回：

一個字典，將 PyTorch API 中的可重寫函數映射到與實際函數具有相同簽名並無條件返回 -1 的 lambda 函數。這些 lambda 函數對於測試定義 __torch_function__ 的類型的 API 覆蓋率非常有用。

返回類型：

字典[可調用，可調用]

返回一個包含所有可覆蓋函數的虛擬覆蓋的字典

例子

>>> import inspect
>>> my_add = torch.overrides.get_testing_overrides()[torch.add]
>>> inspect.signature(my_add)
<Signature (input, other, out=None)>