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


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


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

tf.mod()函數返回元素的除法餘數。

運算:floor(x /y)* y + mod(x,y)= x。

注意:它支持廣播。

用法:



tf.mod(x, y).

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

  • x:它是張量。
  • y:它是張量。它必須與x具有相同的類型。

返回值:張量。與x具有相同的類型。

範例1:我們還公開了tf.mod(),它具有與此操作相同的簽名,並且斷言a和b具有相同的形狀(不廣播)。

Javascript


// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor
const x = tf.tensor([9, 12, 3, 20, 7]);
const y = tf.tensor([2, 2, 9, 4, 2]);
  
tf.mod(x,y).print();

輸出:

Tensor
    [1, 0, 3, 0, 1]

範例2:在操作中將數組和標量值組合在一起的最簡單的廣播示例:

Javascript


// Importing the tensorflow library
import * as tf from "@tensorflow/tfjs"
  
// Broadcast a mod b.
const x = tf.tensor([2, 4, 5, 8]);
const y = tf.scalar(5);
  
x.mod(y).print();

輸出:

Tensor
    [2, 4, 0, 3]

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

相關用法


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