Python PyTorch is_tensor_like用法及代码示例

本文简要介绍python语言中 torch.overrides.is_tensor_like 的用法。

用法:

torch.overrides.is_tensor_like(inp)

如果 passed-in 输入是 Tensor-like，则返回 True。

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

例子

张量的子类通常是Tensor-like。

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

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

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

但是，它们可以通过实现 __torch_function__ 来制作 Tensor-like。

>>> class TensorLike:
...     @classmethod
...     def __torch_function__(cls, func, types, args, kwargs):
...         return -1
>>> is_tensor_like(TensorLike())
True