之前遇到转为tensor转化为浮点型的问题,今天整理下,我只讲几个我常用的,如果有更好的方法,欢迎补充
一、torch.tensor
1.首先讲下torch.tensor,默认整型数据类型为torch.int64,浮点型为torch.float32

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

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

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

补充:还有一种方法通过numpy数组定义数据类型,再转化为tensor,这个方法不多讲了。
文章介绍了在PyTorch中使用torch.tensor和torch.FloatTensor进行数据类型转换的方法,特别是如何将数据转为浮点型。torch.tensor通过dtype参数定义数据类型,而torch.FloatTensor用于将列表或数组转换为32位浮点型Tensor,同时提到了使用numpy数组定义数据类型然后转化为tensor的方式。
815

被折叠的 条评论
为什么被折叠?



