站点图标 AI技术聚合

Pytorch数据类型转换(torch.tensor,torch.FloatTensor)

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

一、torch.tensor

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

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

二、torch.FloatTensor

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

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

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

文章出处登录后可见!

已经登录?立即刷新
退出移动版