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

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

一、torch.tensor

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

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

二、torch.FloatTensor

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

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

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

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年8月9日
下一篇 2023年8月9日

相关推荐