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

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

tf.layers.convLstm2d() 函數用於創建一個 ConvRNN2D 層，該層由一個 ConvLSTM2DCell 組成，ConvLSTM2D 的 apply 方法對一係列輸入進行操作。輸入的形狀至少需要是 4D，第一個維度是時間步長。 convLstm2d 是卷積 LSTM 層。

用法:

tf.layers.convLstm2d( args )

參數：

• args: 它指定給定的配置對象：
  • activation: 這是要使用的激活函數。默認為雙曲正切。如果傳遞 null，則不會使用激活函數。
  • useBias: 它指定是否使用偏差向量。
  • kernelInitializer: 它是核權重矩陣的初始化器，用於輸入的線性變換。
  • recurrentInitializer: 它是recurrentInitializer權重矩陣的Initializer，用於循環狀態的線性變換。
  • biasInitializer: 它是偏置向量的初始值設定項。
  • kernelRegularizer: 它是應用於核權重矩陣的正則化函數。
  • recurrentRegularizer: 它是應用於 recurrentKernel 權重矩陣的正則化函數。
  • biasRegularizer:  它是應用於偏置向量的正則化函數。
  • kernelConstraint: 它是應用於核權重矩陣的約束函數。
  • recurrentConstraint:  它是應用於 recurrentKernel 權重矩陣的約束函數。
  • biasConstraint: 它是應用於偏置向量的約束函數。
  • dropout: 它是 0 到 1 之間的數字，指示輸入線性變換要丟棄的單位。
  • recurrentDropout: 它是降低循環狀態線性變換的分數。
  • dropoutFunc: 這用於測試 DI。
  • inputShape: 它應該是一個數字數組。該字段用於創建一個輸入層，該輸入層用於插入到該層之前。
  • batchInputShape: 它應該是一個數字數組。如果提供 inputShape 和此字段作為創建用於插入到該圖層之前的輸入圖層的參數，則將使用此字段。
  • batchSize: 它應該是一個數字。在沒有batchInputShape的情況下，該字段用於使用inputShape創建batchInputShape。 batchInputShape：[batchSize，...inputShape]。
  • dtype: 如果該層用作輸入層，則該字段用作該層的數據類型。
  • name: 它應該是字符串類型。該字段定義該層的名稱。
  • trainable: 它應該是布爾值。該字段定義該層的權重是否可以通過擬合進行訓練。
  • weights: 這應該是一個定義該層初始權重值的張量。
  • inputDType: 這是用於舊版支持的數據類型。
  • recurrentActivation:它指定將用於循環步驟的激活函數。該參數的默認值為硬 sigmoid。
  • unitForgetBias: 它應該是布爾值。它用於在初始化時將遺忘門的偏差加 1。
  • implementation: 它指定了實現模式。它可以是 1 或 2。為了獲得卓越的性能，建議實施。
  • returnSequences:  它應該是布爾值。它定義了這兩個天氣最後輸出或完整序列在輸出序列中返回。
  • returnState: 它應該是布爾值。它定義輸出中返回的最後狀態。
  • goBackwards: 它應該是布爾值。它定義是否以相反和相反的順序處理輸入。
  • stateful: 它應該是布爾值。它定義是否使用批次的最後狀態作為當前批次的初始狀態。
  • unroll: 它應該是布爾值。它告訴是否使用展開的網絡，否則使用符號循環。
  • inputDim: 它應該是一個數字。當該層用作模型的第一層時使用此選項。它定義了層中輸入的維度。
  • inputLength: 它應該是一個數字。當輸入數據的順序恒定時，應使用此字段。
  • cell: 它應該是 RNN 單元的實例或 RNN 單元實例的數組。
  • filters: 它是多個濾波器的卷積。
  • kernelSize: 它是卷積窗口將具有的維度。
  • strides: 它是每個維度上卷積層的步長。
  • padding: 它應該是 '有效’，‘same’，和‘causal’。它定義了填充模式。
  • dataFormat: 它定義了數據的格式，它告訴輸入中維度的順序。
  • dilationRate: 卷積層在每個維度上應擴張到的擴張率。

返回：它返回 ConvLSTM2D。

示例 1：在此示例中，我們將僅在一個輸入中操作 apply 方法。

// Config for input layer
const bSize = 3;
const sLength = 4;
const size = 7;
const channels = 2;
const inputShape = [bSize, sLength, size, size, channels];
const input = tf.zeros(inputShape);
// Config for convLstm2d 
const filters = 4;
const kernelSize = 4;
const layer = tf.layers.convLstm2d({filters, kernelSize});
const output = layer.apply(input);
// Printing Shape of our output
console.log(JSON.stringify(output.shape));

輸出：

[3,4,4,4]

示例 2：

import * as tf form "@tensorflow/tfjs"
const filters = 3;
const kernelSize = 3;
const batchSize = 4;
const sequenceLength = 2;
const size = 5;
const channels = 3;
const Input = tf.input({shape: [2, 3, 5, 4]});
const Data = tf.ones([1, 2, 3, 5, 4]);
const inputShape = [batchSize, 
    sequenceLength, size, size, channels];
const input = tf.ones(inputShape);
const layer = tf.layers.convLstm2d({filters, kernelSize});
const output = layer.apply(Input);
const model = tf.model({inputs: Input, outputs: output});
model.predict(Data).print();

輸出：

Tensor
    [[[[0.2382042, 0.3073192, -0.0167814],
       [0.2362143, 0.3104056, -0.0188109],
       [0.2277206, 0.3124302, -0.0153713]]]]

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