评价此页

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