- rank: 维度
- shape:类型
- argmax:取最大值的索引号
- tf.argmax(arr, 0):取每一列最大值的索引号
- tf.argmax(arr, 1):取每一行最大值的索引号
tf.argmax的用法
猜你喜欢
转载自blog.csdn.net/Miracle_520/article/details/85264612
今日推荐
周排行