torch.is_tensor

torch.is_tensor(obj, /)

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

请注意，此函数只是执行 isinstance(obj, Tensor)。对于使用 mypy 进行类型检查，使用 isinstance 检查更好，并且更明确 - 因此建议使用它而不是 is_tensor。

参数

obj (对象) – 要测试的对象

返回类型

typing_extensions.TypeGuard[张量]

示例

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