Python PyTorch is_tensor_method_or_property用法及代碼示例

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

用法:

torch.overrides.is_tensor_method_or_property(func)

如果傳入的函數是屬於 torch.Tensor 的方法或屬性的處理程序，則返回 True，如傳遞給 __torch_function__ 。

注意

對於屬性，必須傳入它們的__get__ 方法。

這可能是需要的，特別是由於以下原因：

1. 方法/屬性有時不包含 __module__ 插槽。

2. 他們要求第一個 passed-in 參數是 torch.Tensor 的一個實例。

例子

>>> is_tensor_method_or_property(torch.Tensor.add)
True
>>> is_tensor_method_or_property(torch.add)
False