Tensorflow.js tf.constraints.nonNeg()用法及代码示例

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

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

用法：

 tf.constraints.nonNeg()

Parameters:

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

相关用法

注：本文由纯净天空筛选整理自abhinavjain194大神的英文原创作品 Tensorflow.js tf.constraints.nonNeg() Function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。