Tensorflow.js tf.topk()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf.topk()函数以及最后一个维度也用于查找k个最大条目的值和索引。

用法:

tf.topk (x, k?, sorted?)

参数：

• x:一维或更高的tf张量，最后一个尺寸至少为k。
• k:它是要查找的元素数量。
• sorted:它是布尔值。如果为true，则结果k个元素将按值降序排列。

返回值：{值：tf.Tensor，索引：tf.Tensor}。它返回一个带有两个包含值和索引的张量的对象。

范例1：

const tf = require("@tensorflow/tfjs")
  
// Creating a 2d tensor
const a = tf.tensor2d([[1, 20, 3], [4, 3, 1], [8, 9, 10]]);
const {values, indices} = tf.topk(a);
  
// Printing the values and indices
values.print();
indices.print();

输出：

Tensor
    [[20],
     [4 ],
     [10]]
Tensor
    [[1],
     [0],
     [2]]

范例2：在此示例中，我们将提供参数k以获得最大的k个条目。

const tf = require("@tensorflow/tfjs")
  
// Creating a 2d tensor
const a = tf.tensor2d([[1, 20, 3], [4, 3, 1], [8, 9, 10]]);
const {values, indices} = tf.topk(a, 3);
  
// Printing the values and indices
values.print();
indices.print();

输出：

当我们传递k = 3时，我们得到3个最大值。

Tensor
    [[20, 3, 1],
     [4 , 3, 1],
     [10, 9, 8]]
Tensor
    [[1, 2, 0],
     [0, 1, 2],
     [2, 1, 0]]

参考： https://js.tensorflow.org/api/latest/#topk