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