栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

matplotlib plt.hist用torch tensor作输入作图极慢

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

matplotlib plt.hist用torch tensor作输入作图极慢

背景

先用torch的框架算出来了一批数据,想要看看他的直方图,结果直接放进去,plt作图慢道怀疑人生。

t=cos_dist_with_tag[keep_real_term_mask]
cos_dist=t[:,keep_real_term_mask] 

cos_dist_f=cos_dist.flatten()# torch.tensor变量

plt.figure(figsize=(16,8))
plt.subplot(1,2,1)
plt.title('cos unwhite dist')
plt.hist(cos_dist_f,bins=100,label='unwhite')
plt.legend()
原因

因为plt库不能直接转换tensor的变量!反之会统计成torch.tensor(0.9)即便是0.9一个值。
要先把tensor向量转化成至少numpy数据,就可以秒出结果。
tensor变量已经有方法叫.numpy(),可以直转。

解决
t=cos_dist_with_tag[keep_real_term_mask]
cos_dist=t[:,keep_real_term_mask] 

cos_dist_f=cos_dist.flatten().numpy()# torch.tensor变量 转化numpy

plt.figure(figsize=(16,8))
plt.subplot(1,2,1)
plt.title('cos unwhite dist')
plt.hist(cos_dist_f,bins=100,label='unwhite')
plt.legend()
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/926229.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号