• 一、torch.tensor
  • 二、torch.FloatTensor

之前遇到转为tensor转化为浮点型的问题,今天整理下,我只讲几个我常用的,如果有更好的方法,欢迎补充

一、torch.tensor

1.首先讲下torch.tensor,默认整型数据类型为torch.int64,浮点型为torch.float32

2.这是我认为平常最爱用的转数据类型的方法,可以用dtype去定义数据类型

二、torch.FloatTensor

1.这个函数不要乱用,首先它可以将变量转化为浮点型32位,这里注意此时的变量类型为列表,或数组等,此时参数为单个变量

2.当函数参数为整形时,表示生成矩阵的维度,此时参数可以为多个变量

补充:还有一种方法通过numpy数组定义数据类型,再转化为tensor,这个方法不多讲了。