📌  相关文章
📜  Tensorflow.js tf.initializers.constant() 方法

📅  最后修改于: 2022-05-13 01:56:39.439000             🧑  作者: Mango

Tensorflow.js tf.initializers.constant() 方法

Tensorflow.js 中的初始化器用于初始化内核、权重和偏差的起始值。 tf.initializers.constant()是继承自 Initializer 基类的初始化函数。此函数用于生成初始化为某个常量的值。在这篇文章中,我们将了解 Tensorflow.js 中的tf.initializers.constant()函数。

句法:

tf.initializers.constant(args)

参数: args对象包含以下道具。

  • values:变量中每个元素的值。

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

示例 1:在此示例中,我们将看到 tf.initializers.constant()函数的独立使用。

Javascript
// 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);


Javascript
// 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();


输出:

Constant { value: 7 }

示例 2:在此示例中,我们将在模型创建中使用 constant()函数来初始化内核。

Javascript

// 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