tf.argmax(input, axis=None, name=None, dimension=None)
Returns the index with the largest value across axes of a tensor.
本文详细介绍了 TensorFlow 中的 argmax 函数,该函数能够返回张量沿指定轴的最大值索引。argmax 在机器学习任务中常用于确定预测类别。
tf.argmax(input, axis=None, name=None, dimension=None)
Returns the index with the largest value across axes of a tensor.
353

被折叠的 条评论
为什么被折叠?