张量流中的index()函数?

时间:2017-01-25 16:11:54

标签: tensorflow

tensorflow中有没有办法得到张量值的索引?

e.g。我有一个单热矩阵,我想得到1的坐标。

| 0 0 0 |
| 0 1 0 |  => (1,1)
| 0 0 0 |

优选地,这可以通过例如函数来完成。 tensor.index(binary_function)

1 个答案:

答案 0 :(得分:2)

您可以使用tf.where

例如,

NameValueCollection