評價此頁

torch.is_tensor#

torch.is_tensor(obj, /)[source]#

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

請注意,此函式僅執行 isinstance(obj, Tensor)。使用該 isinstance 檢查對於 mypy 的型別檢查更佳,並且更明確——因此建議使用它而不是 is_tensor

引數

obj (object) – 要測試的物件

返回型別

TypeIs[Tensor]

示例

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