Shortcuts

torch.is_tensor

torch.is_tensor(obj)[源代码]

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

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

Parameters

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

示例:

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