pytorch中函数tensor.numpy()的数据类型实例分析(pytorch,tensor.numpy(),开发技术)

时间:2024-04-29 17:45:22 作者 : 石家庄SEO 分类 : 开发技术
  • TAG :

函数tensor.numpy()的数据类型

今天写代码的时候,要统计一下标签数据里出现的类别总数和要分类的分类数是不是一致的。

我的做法是把tensor类型的数据转变成list,然后用Counter函数做统计。

代码如下:

fromcollectionsimportCounterList_counter=Counter(List1)#List1就是待统计的数据,是一维的列表。生成的List_counter是一个字典,键是数据,#对应的值是数据出现的频率

在做这个统计的时候,突然发现,我的数据是float的类型,这是不应该出现的,因为标签数据在处理的时候都是整型数据。

经过一番查找后,发现是tensor.numpy()返回值数据类型的原因。这个函数的返回值是float类型的

tensor数据和numpy数据转换中注意的一个问题

在pytorch中,把numpy.array数据转换到张量tensor数据的常用函数是torch.from_numpy(array)或者torch.Tensor(array)

第一种函数更常用,然而在pytorch0.4中已经舍弃了这种函数

下面一个简单的编程实验说明这两种方法的区别

实验在pytorch0.4框架下进行

pytorch中函数tensor.numpy()的数据类型实例分析

运行程序之后,结果是

pytorch中函数tensor.numpy()的数据类型实例分析

可以看出修改数组a的元素值,张量b的元素值也改变了,但是张量c却不变。

修改张量c的元素值,数组a和张量b的元素值都不变。

这说明torch.from_numpy(array)是做数组的浅拷贝,torch.Tensor(array)是做数组的深拷贝

 </div> <div class="zixun-tj-product adv-bottom"></div> </div> </div> <div class="prve-next-news">
本文:pytorch中函数tensor.numpy()的数据类型实例分析的详细内容,希望对您有所帮助,信息来源于网络。
上一篇:Vue如何实现嵌套菜单组件下一篇:

5 人围观 / 0 条评论 ↓快速评论↓

(必须)

(必须,保密)

阿狸1 阿狸2 阿狸3 阿狸4 阿狸5 阿狸6 阿狸7 阿狸8 阿狸9 阿狸10 阿狸11 阿狸12 阿狸13 阿狸14 阿狸15 阿狸16 阿狸17 阿狸18