torch.is_tensor() 和 isinstance()

1 torch.is_tensor()

用来判断对象是不是 tensor, 是tensor的话返回true，否则返回false

torch.is_tensor(obj)

示例

import torch

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

>>True

2 isinstance()

isinstance(obj, type)

这个方法可以检查任何类型，如果检查出obj是type类型返回true，否则返回false

import torch

a = torch.tensor([[1, 1], [2, 2]])
isinstance(a, Tensor)

>>name 'Tensor' is not defined

注意 isinstance() 里面的 tensor 类型不是 Tensor, 而是 torch.Tensor

import torch

a = torch.tensor([[1, 1], [2, 2]])
print(type(a))
isinstance(a, torch.Tensor)

>><class 'torch.Tensor'>
>>True