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