torch.is_tensor 简介

torch.is_tensor(obj),此方法很直观,如果obj是tensor的话返回true,否则返回false。与此方法对应的另一个方法是:isinstance(obj, Tensor)。

需要注意的是,torch.is_tensor(obj)是torch的一个方法,而isinstance(obj, Tensor)是python自带的一个方法,这两个是等价的。当然isinstance(obj, type)这个方法可以检查任何类型,如果检查出obj是type类型返回true,否则返回false。

torch.is_tensor 应用举例

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