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

tf.layers.dense() 是 Tensorflow.js 庫的內置函數。此函數用於創建全連接層，其中每個輸出都取決於每個輸入。

用法:

tf.layers.dense(args)

參數：此函數將 args 對象作為參數，它可以具有以下屬性：

units:它是一個正數，定義了輸出空間的維度。
activation:指定使用哪個激活函數。
useBias:指定是否應用偏差。
kernelInitializer:指定用於密集內核權重矩陣的初始值設定項。
biasInitializer:指定該層的偏置向量。
inputDim:將輸入形狀定義為 [inputDim]。
kernelConstraint:指定內核的約束。
biasConstraint:偏置向量的特定約束。
kernelRegularizer:指定應用於密集內核權重矩陣的正則化函數。
biasRegularizer:指定應用於偏置向量的正則化函數。
activityRegularizer:指定應用於激活的正則化函數。
inputShape:如果定義了此參數，它將創建另一個輸入層以插入到該層之前。
batchInputShape:如果定義了此參數，它將創建另一個輸入層以插入到該層之前。
batchSize:用於構造batchInputShape(如果尚未指定)。
dtype:指定該圖層的數據類型。該參數的默認值為 ‘float32’。
name:指定該層的名稱。
trainable:指定該層的權重是否通過擬合更新。
weights:指定圖層的初始權重值。
inputDType：用於表示inputDType，其值可以是‘float32’或‘int32’或‘bool’或‘complex64’或‘string’。

返回值：它返回一個 Dense 對象。

範例1：

Javascript


import * as tf from "@tensorflow/tfjs"
  
// Create a new dense layer
const denseLayer = tf.layers.dense({
   units:2,
   kernelInitializer:'heNormal',
   useBias:true
});
    
const input = tf.ones([2, 3]);
const output = denseLayer.apply(input);
    
// Print the output
output.print()

輸出：

範例2：

Javascript


import * as tf from "@tensorflow/tfjs"
  
// Create a new dense layer
const denseLayer = tf.layers.dense({
   units:3,
   kernelInitializer:'heNormal',
   useBias:false
});
    
const input = tf.ones([2, 3, 3]);
const output = denseLayer.apply(input);
    
// Print the output
output.print()

輸出：

範例3：

Javascript


import * as tf from "@tensorflow/tfjs"
    
// Create a new dense layer
const denseLayer = tf.layers.dense({
   units:3,
   kernelInitializer:'ones',
   useBias:false
});
    
const input = tf.ones([2, 3, 3]);
const output = denseLayer.apply(input);
    
// Print the output
output.print()

輸出：

範例4：

Javascript


import * as tf from "@tensorflow/tfjs"
  
// Create a new dense layer
const denseLayer = tf.layers.dense({
   units:3,
   kernelInitializer:'randomUniform',
   useBias:false
});
    
const input = tf.ones([2, 3, 3]);
const output = denseLayer.apply(input);
    
// Print the output
output.print()

輸出：

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

相關用法

注：本文由純淨天空篩選整理自abhinavjain194大神的英文原創作品 Tensorflow.js tf.layers.dense() Function。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。