Python PyTorch is_tensor用法及代码示例

本文简要介绍python语言中 torch.is_tensor 的用法。

用法:

torch.is_tensor(obj)

参数：

obj(对象) -测试对象

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

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

例子：

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