Tensorflow.js tf.constraints.nonNeg()函数

Tensorflow.js 是一个由谷歌开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

tf.constraints.nonNeg()函数我们用来创建一个 nonNeg 约束。 nonNeg是一个非负权重约束。它继承自约束类。约束是层的属性。

句法：

tf.constraints.nonNeg() 

参数：

• w：指定输入权重变量。它是一个可选参数。

返回值：返回 tf.constraints.Constraint。

示例 1：

Javascript

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Use nonNeg() function
const constraint = tf.constraints.nonNeg( )
 
// Print
console.log(constraint)

输出

{}

示例 2：在此示例中，我们将使用 nonNeg 约束创建一个密集层，并将形成的层应用于张量。

Javascript

// Importing the tensorflow.Js library
import * as tf from "@tensorflow/tfjs"
 
// Create nonNeg constarint using nonNeg() function
const constraint = tf.constraints.nonNeg()
 
// Create a new dense layer using nonNeg constraint
const denseLayer = tf.layers.dense({
    units: 4,
    kernelInitializer: 'heNormal',
    kernelConstraint: constraint ,
    biasConstraint: constraint ,
    useBias: true
});
 
// Create input
const input = tf.ones([2, 2]);
 
// Apply denseLayer to input
const output = denseLayer.apply(input);
 
// Print the output
output.print()

输出

Tensor
    [[-0.7439224, -1.3572885, -1.2860565, 1.3913929],
     [-0.7439224, -1.3572885, -1.2860565, 1.3913929]]

参考： https://js.tensorflow.org/api/1.0.0/#constraints.nonNeg