pytorch 张量 数据类型检测

#引入torch库
import torch

#随机生成生成均值为0，方差为1的2X3矩阵
a = torch.randn(2, 3)

#返回a的类型
b = a.type()

#检验a的数据是不是torch.FloatTensor
c = isinstance(a, torch.FloatTensor)
 
print(type(a))
print(a)
print(b)
print(c)

结果为：

<class 'torch.Tensor'>
tensor([[-1.1685, -1.8951,  1.2218],
        [ 0.8610, -1.0949,  1.4067]])
torch.FloatTensor
True

同一个tensor部署在CPU和GPU上是不一样的

import torch
data = torch.randn(2, 3)#默认的数据类型为FloatTensor

d = isinstance(data, torch.cuda.FloatTensor)

print(d)

输出为：False

要想实现CPU到GPU需要一个数据的搬运，在进行数据比较，即为真

import torch
data = torch.randn(2, 3)
data = data.cuda()
d = isinstance(data, torch.cuda.FloatTensor)
print(d)

#输出为：
True