torch.is_tensor

torch.is_tensor(obj, /)[源代码]

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

请注意，此函数只是执行isinstance(obj, Tensor)。使用isinstance检查对于与mypy进行类型检查更有效，并且更加明确——因此建议用isinstance代替is_tensor。

参数

obj (对象) – 需要测试的目标对象

返回类型

TypeGuard[\Tensor]

示例：

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