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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。