Pytorch数据类型转换（torch.tensor，torch.FloatTensor）

• Python

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

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

一、torch.tensor

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

在这里插入图片描述

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

在这里插入图片描述

二、torch.FloatTensor

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

在这里插入图片描述

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

在这里插入图片描述

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

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://net2asp.com/13a609baeb.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐