Tensorflow.js tf.oneHot()用法及代碼示例

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。

tf.oneHot()函數用於創建one-hot tf.Tensor。由索引表示的位置將值設為1(默認值)，也稱為onValue，而所有其他位置將值設為0(默認值)，也稱為offValue。

用法:

tf.oneHot (indices, depth, onValue, offValue)

參數：此函數接受三個參數，如下所示：

• indices它可以使用dtype int32的tf.Tensor(TypedArray或Array)索引。
• depth深度的數據類型是數字。它用於表示一個熱尺寸的深度。
• onValueonValue的數據類型是數字。當索引與位置匹配時，用於填寫輸出。它是一個可選參數。
• offValueoffValue的數據類型是數字。當索引與位置不匹配時，用於填寫輸出。這也是一個可選參數。

返回：它返回一個tf.Tensor。

範例1：

// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Use of oneHot function.
var val = tf.oneHot(tf.tensor1d([0,1,2], 'int32'), 3);;
  
// Printing the tensor
val.print()

輸出：

Tensor
    [[1, 0, 0],
    [0, 1, 0],
    [0, 0, 1]]

範例2：

// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
  
// Creating and intializing a new variable 
var val = tf.oneHot(tf.tensor1d([0,1,2], 'int32'), 3,9,-1);
  
// Printing the tensor
val.print()

輸出：

Tensor
    [[9 , -1, -1],
    [-1, 9 , -1],
    [-1, -1, 9 ]]

參考：https://js.tensorflow.org/api/latest/#oneHot