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

Tensorflow.js 中的初始化器用於初始化 kernel、weights 和 baises 的起始值。 tf.initializers.constant() 是從 Initializer 基類繼承的初始化函數。此函數用於生成初始化為某個常量的值。在這篇文章中，我們將了解 Tensorflow.js 中的 tf.initializers.constant() 函數。

用法:

tf.initializers.constant(args)

參數：args 對象包含以下道具。

• values:變量中每個元素的值。

返回值：它返回 tf.initializers.Initializer。

範例1：在這個例子中，我們將看到 tf.initializers.constant() 函數的獨立使用。

// Importing the tensorflow.js library
const tf = require("@tensorflow/tfjs")
 
// Use  tf.initializers.constant() function
var initializer = tf.initializers.constant({ value:7, })
 
// Print the value of constant
console.log(initializer);

輸出：

Constant { value:7 }

範例2：在這個例子中，我們將在模型創建中使用 constant() 函數來初始化內核。

// Importing the tensorflow.js library
const tf = require("@tensorflow/tfjs")
 
// Using tf.initializers.constant() function
var initializer = tf.initializers.constant({ value:7, })
 
// Create model
const model = tf.sequential();
 
// Add layer and initialize the kernel
model.add(tf.layers.dense({
    units:3,
    activation:'softmax',
    kernelInitializer:initializer,
    inputShape:[2]
}));
 
// Print the summary
model.summary();

輸出：

Layer (type)                 Output shape              Param #   
=================================================================
dense_Dense1 (Dense)         [null,3]                  9
=================================================================
Total params:9
Trainable params:9
Non-trainable params:0
_________________________________________________________________

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