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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。