tf.nn.top_k ￡神魔★判官ぃ 2021-12-20 15:13 215阅读 0赞 **评估操作对于测量神经网络的性能是有用的。 由于它们是不可微分的，所以它们通常只是被用在评估阶段** ### `tf.nn.top_k(input, k, name=None)` ### 这个函数的作用是返回 input 中每行最大的 k 个数，并且返回它们所在位置的索引。 输入参数： input: 一个张量，数据类型必须是以下之一：float32、float64、int32、int64、uint8、int16、int8。数据维度是 batch\_size 乘上 x 个类别。 k: 一个整型，必须 >= 1。在每行中，查找最大的 k 个值。 name: 为这个操作取个名字。 **输出参数：** **一个元组 Tensor** ，数据元素是 (values, indices)，具体如下： values: 一个张量，数据类型和 input 相同。数据维度是 batch\_size 乘上 k 个最大值。 indices: 一个张量，数据类型是 int32 。每个最大值在 input 中的索引位置。 \--------------------- 作者：Never-Giveup 来源：CSDN 原文：https://blog.csdn.net/qq\_36653505/article/details/81105894 版权声明：本文为博主原创文章，转载请附上博文链接！ input = tf.constant(np.random.rand(3,4)) k = 2 output = tf.nn.top\_k(input, k) with tf.Session() as sess: print(sess.run(input)) print(sess.run(output)) [[0.61950464 0.34474213 0.79035374 0.15015998] [0.17963278 0.30331155 0.9208411 0.90382958] [0.20007082 0.89997606 0.03721232 0.24253472]] TopKV2(values=array([[0.79035374, 0.61950464], [0.9208411 , 0.90382958], [0.89997606, 0.24253472]]), indices=array([[2, 0], [2, 3], [1, 3]], dtype=int32)) 转载于:https://www.cnblogs.com/hapyygril/p/10824158.html
还没有评论，来说两句吧...