torch.is_tensor

torch.is_tensor(obj, /)

如果 obj 是 PyTorch 張量，則返回 True。

請注意，此函式僅執行 isinstance(obj, Tensor) 操作。使用該 isinstance 檢查更適合 mypy 的型別檢查，並且更明確——因此建議使用它代替 is_tensor。

引數

obj (物件) – 待測試物件

返回型別

typing_extensions.TypeIs[張量]

示例

>>> x = torch.tensor([1, 2, 3])
>>> torch.is_tensor(x)
True