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