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