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


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

Tensorflow.js是Google開發的開源庫,用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。

tf.maximum()函數用於按元素返回兩個指定張量的最大值。它支持廣播。

用法:

tf.maximum (a, b)

參數:此函數接受兩個參數,如下所示:

  • a:第一個指定的張量。
  • b:第二個指定的張量。它必須具有與“a”相同的數據類型。

返回值:它按元素返回兩個指定張量“a”和“b”的最大值。



範例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing two tensors
const a = tf.tensor1d([3, 5, 7, 9]);
const b = tf.tensor1d([1, 6, 8, 2]);
  
// Calling the .maximum() function
a.maximum(b).print();

輸出:

Tensor
   [3, 6, 8, 9]

範例2:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Using two tensors as the parameter
// for the .maximum() function
tf.tensor1d([3, 5, 7, 9]).maximum(tf.tensor1d([6])).print();

輸出:

Tensor
   [6, 6, 7, 9]

範例3:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Broadcasting maximum a with b
const a = tf.tensor1d([1, 2, 7, 3, 4]);
const b = tf.scalar(4);
  
// Calling the .maximum() function
a.maximum(b).print();

輸出:

Tensor
   [4, 4, 7, 4, 4]

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

相關用法


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