问numpy中矩阵的One-hot表示
EN

Stack Overflow用户

提问于 2017-01-07 03:49:29

回答 1查看 496关注 0票数 2

在3d张量中，从一个值矩阵到同一事物的一个热门表示的最简单/最智能的方法是什么？例如，如果矩阵是张量中argmax之后的索引，如下所示：

indices=numpy.argmax(mytensor,axis=2)

其中张量是3Dx，y，z，索引自然是x，y，现在你想要得到一个3Dx，y，z张量，在axis=2中用1代替最大值，在其他地方用0代替。

我知道向量到1-热矩阵的答案，但这是对1-热张量的矩阵。

回答已采纳

发布于 2017-01-07 03:51:33

使用broadcasting的完美设置之一-

indices[...,None] == np.arange(mytensor.shape[-1])

如果需要在0s和1s的in中，追加.astype(int)

票数 3

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/41513252

复制

相似问题

问numpy中矩阵的One-hot表示EN