标签: tensorflow
tensorflow中有没有办法得到张量值的索引?
e.g。我有一个单热矩阵,我想得到1的坐标。
| 0 0 0 | | 0 1 0 | => (1,1) | 0 0 0 |
优选地,这可以通过例如函数来完成。 tensor.index(binary_function)
tensor.index(binary_function)
答案 0 :(得分:2)
您可以使用tf.where。
tf.where
例如,
NameValueCollection