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

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

這個tf.layers.average() 函數當多個數組作為輸入提供時，用於計算每個元素的平均值。

用法：

tf.layers.average (inputShape?, batchInputShape?, batchSize?, 
        dtype?, name?, trainable?, weights?, inputDType? )

參數：

輸入形狀：這是一個用於創建輸入層的可選參數，它采用 number 和 null 等值。
批處理輸入形狀：這是一個可選參數，用於在主層之前創建輸入層，它采用 number 和 null 等值。
批量大小：它是用於製作 batchInputShape 的可選參數，它隻接受數字。
數據類型：它是一個可選參數，它代表數據類型。默認情況下，它有 ‘float32’ 並且還支持其他值，如 ‘int32’, ‘bool’ 等。
name:它是一個可選參數，用於定義層的名稱，它接受字符串。
trainable:它是一個可選參數，用於確定提供的輸入層是否更新。它接受布爾值。
weights:它擁有層的起始權重。它也是一個可選參數。
inputDType:它是用於輸入數據類型的可選參數。像 dtype 一樣，它也支持它的所有值。

返回值：它返回平均值。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const input1 = tf.input({shape:[33, 3]});
const input2 = tf.input({shape:[33, 3]});
const averageLayer = tf.layers.average();
const average = averageLayer.apply([input1, input2]);
console.log(JSON.stringify(average.shape));

輸出：

[null, 33, 3]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const input1 = tf.input({shape:[4, 4]});
const input2 = tf.input({shape:[4, 4]});
const input3 = tf.input({shape:[4, 4]});
const input4 = tf.input({shape:[4, 4]});
const input5 = tf.input({shape:[4, 4]});
const input6 = tf.input({shape:[4, 4]});
  
  
  
const averageLayer = tf.layers.average();
const average = averageLayer.apply([input1, 
    input2, input3, input4, input5, input6 ]);
console.log(JSON.stringify(average.shape));

輸出：

[null, 4, 4]

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

相關用法

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