torch.is_tensor

torch.is_tensor(obj, /)

如果 obj 是 PyTorch 张量，则返回 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