Tensor对象是一个任意维度的矩阵,但是一个Tensor中所有元素的数据类型必须一致。
Torch包含的数据类型和普遍的编程语言的数据类型相似,包含浮点型、有符号整型和无符号整型。
这些类型既可以定义在CPU上,也可以定义在GPU上。
一、dtype属性指定数据类型,device指定设备
使用Tensor数据类型时,可以通过dtype属性指定它的数据类型
输出为:
device指定它的设备(CPU或GPU)
Tensor对象是一个任意维度的矩阵,但是一个Tensor中所有元素的数据类型必须一致。
Torch包含的数据类型和普遍的编程语言的数据类型相似,包含浮点型、有符号整型和无符号整型。
这些类型既可以定义在CPU上,也可以定义在GPU上。
使用Tensor数据类型时,可以通过dtype属性指定它的数据类型
输出为:
device指定它的设备(CPU或GPU)