torch.is_storage(obj)函数

用法

torch.is_storage(obj)

该函数用于判断一个对象是否是pytorch的存储（storage)对象。storage是torch中用于存储张量（tensor）数据的底层数据结构
参数

obj:判断对象

结果输出：

如果obj是一个torch存储对象，则函数返回True,否则返回False

例子

>>> a=torch.FloatStorage(6)
>>> print(a)
 -269202751488.0
 3.065200260864105e-41
 2.049904161393319e-31
 4.558704164141495e-41
 8.96831017167883e-44
 0.0
[torch.FloatStorage of size 6]
>>> torch.is_storage(a)
True
>>> b=torch.randn(4,6)
>>> b
tensor([[ 1.4832, -1.5439, -1.8002, -0.9426, -0.0065,  0.6780],
        [-0.7843,  0.5130,  0.9171, -0.3018,  0.1424, -0.6070],
        [-0.8619, -0.0259,  1.2841,  0.6479, -0.8677, -0.2812],
        [ 0.8972, -0.8611,  0.7552, -0.3049, -0.6565,  0.5062]])
>>> torch.is_storage(b)
False