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

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。它還可以幫助開發人員使用JavaScript語言開發ML模型，並可以直接在瀏覽器或Node.js中使用ML。

tf.logSoftmax()函數用於計算對數softmax。

用法：

tf.logSoftmax(logits, axis?) 

參數：

• logits:logits數組
• axis：尺寸softmax將在

返回值：返回tf.Tensor

範例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const a = tf.tensor1d([3, 2, 6]);
  
a.logSoftmax().print();

輸出：

Tensor
    [-3.0658839, -4.0658841, -0.0658839]

範例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const a = tf.tensor2d([2, 4, 6, 1, 3,7], [2, 3]);
  
a.logSoftmax().print();

輸出：

Tensor
    [[-4.1429315, -2.1429317, -0.1429316],
     [-6.0205812, -4.0205812, -0.0205811]]

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