快捷方式

torch.overrides

此模块公开了 __torch_function__ 协议的各种辅助函数。有关 __torch_function__ 协议的更多详细信息,请参阅 扩展 torch Python API

函数

torch.overrides.get_ignored_functions()[source]

返回不能被 __torch_function__ 覆盖的公共函数。

返回

一个包含在 torch API 中公开可用但不能使用 __torch_function__ 覆盖的函数的元组。这主要是由于这些函数的参数都不是张量或张量类。

返回类型

Set[Callable]

示例

>>> torch.Tensor.as_subclass in torch.overrides.get_ignored_functions()
True
>>> torch.add in torch.overrides.get_ignored_functions()
False
torch.overrides.get_overridable_functions()[source]

列出可以通过 __torch_function__ 覆盖的函数

返回

一个字典,将包含可覆盖函数的命名空间映射到该命名空间中可以覆盖的函数。

返回类型

Dict[Any, List[Callable]]

torch.overrides.resolve_name(f)[source]

为传递给 __torch_function__ 的函数获取人类可读的字符串名称

参数

f (Callable) – 要解析名称的函数。

返回

函数的名称;如果求值,它应该返回输入函数。

返回类型

str

torch.overrides.get_testing_overrides()[source]

返回一个包含所有可覆盖函数的虚拟覆盖的字典

返回

一个字典,将 PyTorch API 中的可覆盖函数映射到与真实函数具有相同签名的 lambda 函数,这些函数无条件地返回 -1。这些 lambda 函数对于测试定义了 __torch_function__ 的类型的 API 覆盖范围非常有用。

返回类型

Dict[Callable, Callable]

示例

>>> import inspect
>>> my_add = torch.overrides.get_testing_overrides()[torch.add]
>>> inspect.signature(my_add)
<Signature (input, other, out=None)>
torch.overrides.handle_torch_function(public_api, relevant_args, *args, **kwargs)[source]

使用 __torch_function__ 覆盖检查实现函数。

有关 C++ 实现中此函数的等效项,请参阅 torch::autograd::handle_torch_function。

参数
  • public_api (function) – 公共 torch API 公开的函数,最初调用方式类似于 public_api(*args, **kwargs),现在正在对该函数的参数进行检查。

  • relevant_args (iterable) – 要检查 __torch_function__ 方法的参数的 iterable。

  • args (tuple) – 最初传递给 public_api 的任意位置参数。

  • kwargs (tuple) – 最初传递给 public_api 的任意关键字参数。

返回

根据需要调用 implementation__torch_function__ 方法的结果。

返回类型

object

:raises TypeError : 如果未找到实现。

示例

>>> def func(a):
...     if has_torch_function_unary(a):
...         return handle_torch_function(func, (a,), a)
...     return a + 0
torch.overrides.has_torch_function()

检查可迭代元素中的 __torch_function__ 实现,或检查是否启用了 __torch_function__ 模式。将精确的 TensorParameter 视为不可分派。使用它来保护对 handle_torch_function() 的调用;不要用它来测试是否为 Tensor 类,而是使用 is_tensor_like()。:param relevant_args: 要检查 __torch_function__ 方法的可迭代元素或参数。:type relevant_args: iterable

返回

如果 relevant_args 的任何元素具有 __torch_function__ 实现,则为 True,否则为 False。

返回类型

bool

另请参阅

torch.is_tensor_like

检查某个东西是否为 Tensor 类,包括精确的 Tensor

torch.overrides.is_tensor_like(inp)[source]

如果传入的输入为 Tensor 类,则返回 True

目前,只要输入类型上存在 __torch_function__ 属性,就会发生这种情况。

示例

Tensor 的子类通常是 Tensor 类。

>>> class SubTensor(torch.Tensor): ...
>>> is_tensor_like(SubTensor([0]))
True

内置类型或用户类型通常不是 Tensor 类。

>>> is_tensor_like(6)
False
>>> is_tensor_like(None)
False
>>> class NotATensor: ...
>>> is_tensor_like(NotATensor())
False

但是,可以通过实现 __torch_function__ 使它们成为 Tensor 类。

>>> class TensorLike:
...     @classmethod
...     def __torch_function__(cls, func, types, args, kwargs):
...         return -1
>>> is_tensor_like(TensorLike())
True
torch.overrides.is_tensor_method_or_property(func)[source]

如果传入的函数是 torch.Tensor 的方法或属性的处理程序,则返回 True,如传递给 __torch_function__

注意

对于属性,必须传入其 __get__ 方法。

特别是因为以下原因,这可能是必需的

  1. 方法/属性有时不包含 __module__ 槽。

  2. 它们要求传入的第一个参数是 torch.Tensor 的实例。

示例

>>> is_tensor_method_or_property(torch.Tensor.add)
True
>>> is_tensor_method_or_property(torch.add)
False
返回类型

bool

torch.overrides.wrap_torch_function(dispatcher)[source]

__torch_function__ 相关的功能包装给定函数。

参数

dispatcher (Callable) – 一个可调用对象,它返回传递给该函数的 Tensor 类可迭代对象。

注意

此装饰器可能会降低代码性能。通常,将代码表达为一系列本身支持 __torch_function__ 的函数就足够了。如果你发现自己处于极少数情况,例如你正在包装一个低级库并且还需要它适用于 Tensor 类,那么可以使用此函数。

示例

>>> def dispatcher(a):  # Must have the same signature as func
...     return (a,)
>>> @torch.overrides.wrap_torch_function(dispatcher)
>>> def func(a):  # This will make func dispatchable by __torch_function__
...     return a + 0

文档

访问 PyTorch 的综合开发人员文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源