Question

假设我有一个恒定的张量[[0,0,1,1,2,0][0,1,0,0,0,2]]。

[0,0,1,1,2,0]中最后一次出现最大值的索引为4，[0,1,0,0,0,2]中最后一次出现最大值的索引为5。

所以我想得到的是[4, 5]，有什么想法可以做到吗？谢谢。

我正在使用tensorflow 1.9。

Answer 1

只需弄清楚。

 <header class="main__hero">
      <!-- replace list with hamburger icon when screen is smaller than 600px -->
      <a href="javascript:void(0);" class="icon" onclick="menu()">
        <i class="fa fa-bars fa-lg"></i>
      </a>
      <h1>Landing Page </h1>
    </header>
    <!-- vertical menu for screens smaller than 600px -->
    <ul id="navbar__list_2">
      <li><a href="#section1">Section 1</a></li>
      <li><a href="#section2">Section 2</a></li>
      <li><a href="#section3">Section 3</a></li>
      <li><a href="#section4">Section 4</a></li>
    </ul>

输出为：

import tensorflow as tf

a=tf.constant([[0,0,1,1,2,0], [0,1,0,0,0,2]])

max_idx = tf.cast(a.shape[1], tf.int64) - tf.argmax(tf.reverse(a, [1]), axis=1)-1

sess=tf.Session()

sess.run(max_idx)

张量查找张量中最后一次出现最大值的索引

1 个答案: