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

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

tf.signal.stft()函數用於計算信號的Short-time傅立葉變換。

用法：

tf.signal.stft (signal, frameLength, frameStep, 
                     fftLength?, windowFn?) 

參數：

• signal:一維實值張量。
• frameLength:樣品的窗口長度。
• frameStep:要步進的樣本數。
• fftLength:FFT的大小。
• windowFn:具有窗口長度的可調用對象。

返回值：它返回tf.Tensor。

範例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const input = tf.tensor1d([3, 4, 6])
tf.signal.stft(input, 3, 1).print();

輸出：

Tensor
     [[4 + 0j, 0 + -4j, -4 + 0j],]

範例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const input = tf.tensor1d([1, 2, 3, 4, 5, 6, 7])
tf.signal.stft(input, 3, 1).print();

輸出：

Tensor
    [[2 + 0j, 0 + -2j, -2 + 0j],
     [3 + 0j, 0 + -3j, -3 + 0j],
     [4 + 0j, 0 + -4j, -4 + 0j],
     [5 + 0j, 0 + -5j, -5 + 0j],
     [6 + 0j, 0 + -6j, -6 + 0j]]

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