torch.is_tensor

torch.is_tensor(obj, /)

返回 True，如果 obj 是 PyTorch 张量。

注意，此函数只是简单地执行 isinstance(obj, Tensor) 。使用那个 isinstance 检查更适合与 mypy 进行类型检查，并且更明确——因此建议使用它而不是 is_tensor 。

参数:

obj（对象）- 要测试的对象

返回类型:

TypeIs[Tensor]

示例：

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