Ser*_* S. 5 linear-algebra tensorflow
如何在TensorFlow中获取沿选定轴的张量最大元素的索引?
Ser*_* S. 13
对不起,已经想通了.它是:
tf.argmax(input, axis, name=None)
https://www.tensorflow.org/api_docs/python/tf/argmax
归档时间:
9 年,6 月 前
查看次数:
4298 次
最近记录:
7 年,7 月 前