pytorch: 如何优雅的将 int list 转成 one-hot形式

虽然 pytorch 已经升级到 0.2.0 了,但是,貌似依旧没有简单的 api 来帮助我们快速将 int list 转成 one-hot。那么,如何优雅的实现 one-hot 代码呢?

def one_hot(ids, out_tensor):
    """
    ids: (list, ndarray) shape:[batch_size]
    out_tensor:FloatTensor shape:[batch_size, depth]
    """
    if not isinstance(ids, (list, np.ndarray)):
        raise ValueError("ids must be 1-D list or array")
    ids = torch.LongTensor(ids).view(-1,1)
    out_tensor.zero_()
    out_tensor.scatter_(dim=1, index=ids, src=1.)
    # out_tensor.scatter_(1, ids, 1.0)

scatter_ 是什么鬼?

从 value 中拿值,然后根据 dim 和 index 给自己的相应位置填上值

Tensor.scatter_(dim, index, src)
# index: LongTensor
# out[index[i, j], j] = value[i, j] dim=0
# out[i,index[i, j]] = value[i, j]] dim=1 
# index 的 shape 可以不和 out 的 shape 一致
# value 也可以是一个 float 值, 也可以是一个 FloatTensor 
# 如果 value 是 FloatTensor 的话,那么shape 需要和 index 保持一致

参考资料

https://discuss.pytorch.org/t/convert-int-into-one-hot-format/507/3

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Django Scrapy

macbook苹果笔记本蓝牙不可用

1 最好将所有usb设备拔掉 2 关机 :同时按下shift+control+option+power,保持5秒左右 3 重置: 先按下power键,紧接着...

503
来自专栏石奈子的Java之路

原 荐 SpringBoot 2.0 系列0

1474
来自专栏python爬虫日记

windows平台python 2.7环境编译安装zbarlight

类似于前一篇博文,http://www.cnblogs.com/zhongtang/p/7148082.html中描述的情况。

862
来自专栏大数据智能实战

C# webbrowser爬虫中经常碰到的脚本错误弹出窗口的问题解决

在爬虫过程中,经常会碰到脚本错误的提示窗口。 网上有关于这个错误的好多解决办法。 后来终于测试成功。其实现如下:    //## 这边把脚本错误的压制设置为t...

1905
来自专栏计算机视觉

记录要跑的text to image,一步步不要太精细

首先,我们需要准备下载text  to image 资源,我打算跑的代码地址。 ? 要求的版本 然后查看自己的服务器cudn,tensorflow,NLTK,如...

3206
来自专栏進无尽的文章

代码管理| 简易教程之Cocopods的安装和使用

CocoaPods的工作主要是通过ProjectName.xcworkspace来组织的,在打开ProjectName.xcworkspace文件后,发现Xco...

613
来自专栏Java技术栈

Maven精选系列--发布jar包到Nexus私库

Nexus2可以通过管理界面来上传jar包到私库中,而最新的Nexus3却找不到了上传界面,只能通过以下方式来发布到私库。 发布第三方jar包 这种情况是mav...

31811
来自专栏飞雪无情的博客

Android Gradle插件中文指南

最近想把Android项目从Eclipse迁到Android Studio上。原因呢,Eclipse太慢了,而且经常卡死,Android Studio由于And...

812
来自专栏深度学习那些事儿

关于Ubuntu16.04上N卡驱动导致滚动屏幕卡顿问题

屏幕显示正常,但是在打开网页或者进行跑深度学习程序的时候画面会发生一卡一下的情况,严重时出现类似于死机的情况。查看系统monitor并不现实显存和内存爆表,于是...

1976
来自专栏一个默默无闻的工程师的日常

使用 docker 快速部署 ceph

1342

扫码关注云+社区