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