當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Tensorflow.js tf.leakyRelu()用法及代碼示例


Tensorflow.js是一個開放源代碼庫,由Google開發,用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.leakyRelu()函數用於查找指定張量輸入的泄漏校正線性,並按元素進行操作。

用法:

tf.leakyRelu(x, alpha?)

Parameters: 

  • x:它是張量輸入,可以是tf.Tensor,TypedArray或Array類型。
  • alpha:它是數字類型的可選參數。它定義負值的比例因子,默認值為0.2。

返回值:它返回tf.Tensor對象。



範例1:在此示例中,我們定義了一個輸入張量,然後打印泄漏的整流線性值。為了創建輸入張量,我們使用.tensor1d()方法,為了打印輸出,我們使用.print()方法。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const y = tf.tensor1d([4, -4, 38, NaN]);
  
// Calling leakyRelu() method and
// printing output
y.leakyRelu().print();

輸出:

Tensor
    [4, -0.8, 38, NaN]

範例2:在此示例中,所有參數都直接傳遞給LeakyRelu函數。

Javascript


// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input
var val = [3.6, 77, 5.78799797, 'a'];
  
// Calling tensor1d method
const y = tf.tensor1d(val);
  
// Calling leakyRelu() method
var res = tf.leakyRelu(y, 1.7)
  
// printing output
res.print();

輸出:

Tensor
    [3.5999999, 77, 5.7879982, NaN]

參考: https://js.tensorflow.org/api/latest/#leakyRelu

相關用法


注:本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.leakyRelu() Function。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。