#回顾知识
4.1Tensor对象及其运算
Tensor对象是一个任意维度的矩阵,但是一个Tensor中所有元素的数据类型必须一致,它也包含浮点型,有符号整形和无符号整形。(可以定义在CPU上,也可以定义在GPU上)
一 、dype属性指定它的数据类型,device指定它的设备(CPU 或GPU)
1.torch.tensor
2.可以用list构建
a=torch.tensor([[1,2],[3,4]],dtype=torch.float64)
3.用ndarry构建
b = torch.tensor(np.array([[1,2],[3,4]]), dtype=torch.uint8)
print(a) print(b)
展示输出的结果图