Python PyTorch is_tensor用法及代碼示例

本文簡要介紹python語言中 torch.is_tensor 的用法。

用法:

torch.is_tensor(obj)

參數：

obj(對象) -測試對象

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

請注意，此函數隻是在執行 isinstance(obj, Tensor) 。使用 isinstance 檢查更適合使用 mypy 進行類型檢查，並且更明確 - 因此建議使用它而不是 is_tensor 。

例子：

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