torch.is_tensor

torch.is_tensor(obj, /)

如果 obj 是 PyTorch tensor，則傳回 True。

請注意，此函數只是執行 isinstance(obj, Tensor)。 使用該 isinstance 檢查對於使用 mypy 進行類型檢查更好，並且更明確 - 因此建議使用該檢查來代替 is_tensor。

參數

obj (object) – 要測試的物件

傳回類型

typing_extensions.TypeIs[Tensor]

範例

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