tensorflow：一个二维张量：a = [[1,3][2,5]]，怎么在不把它转化成numpy数组的前提下，求出最大值所在的索引啊？argmax函数只能按行或列求索引，有没有办法让我直接得到值5的索引(1,1)呢？
https://cloud.tencent.com/developer/ask/31678