tf.argmax()的用法
tf.argmax()的主要功能是找出最大值并返回索引,注意是索引。下面是主要的内参数tf.argmax( input, axis=None, name=None, dimension=None, output_type=tf.int64)# input代表输入的数组# axis代表按行判断最大值还是按列判断最大值0代表列,1代表行默认为None(0)# name代表操作的名称,默认为None举例说明:import tensorflow as tfi
原创
2020-12-14 09:37:34 ·
3012 阅读 ·
0 评论