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

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

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

一、torch.tensor

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

Pytorch数据类型转换(torch.tensor,torch.FloatTensor)_第1张图片

2.这是我认为平常最爱用的转数据类型的方法,可以用dtype去定义数据类型
Pytorch数据类型转换(torch.tensor,torch.FloatTensor)_第2张图片

二、torch.FloatTensor

1.这个函数不要乱用,首先它可以将变量转化为浮点型32位,这里注意此时的变量类型为列表,或数组等,此时参数为单个变量
Pytorch数据类型转换(torch.tensor,torch.FloatTensor)_第3张图片
2.当函数参数为整形时,表示生成矩阵的维度,此时参数可以为多个变量
Pytorch数据类型转换(torch.tensor,torch.FloatTensor)_第4张图片

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

你可能感兴趣的:(Pytorch,pytorch,python,深度学习)