torch.is_tensor(obj):判断obj是否是tensor类型
import torch
x = torch.randn(1,2)
y = x.numpy()
print(torch.is_tensor(x))
print(torch.is_tensor(y))
True
False
torch.is_storage(obj): 判断obj是否是pytorch storage 对象
从源码来看,storage对象有以下几种,具体什么含义,暂不清楚。有需求再更新。
_storage_classes =