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

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

tf.dot()函數用於計算兩個給定矩陣或向量t1和t2的點積。

用法：

tf.dot(t1, t2);

參數：此函數接受如下所示的參數：

t1:這是點運算的第一個張量。
t2:它是點運算的第二張量。

返回值：它返回兩個給定矩陣或向量t1和t2的點積。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
let geek1 = tf.tensor1d([2, 3]);
let geek2 = tf.tensor2d([[2, 3], [4, 5]]);
let geek3 = tf.tensor2d([[2, 3, 4], [5, 6, 7]]);
  
let a =geek1.dot(geek2);
let b =geek2.dot(geek1);
let c =geek2.dot(geek3);
  
a.print();
b.print();
c.print();

輸出：

Tensor
    [16, 21]
Tensor
    [13, 23]
Tensor
    [[19, 24, 29],
     [33, 42, 51]]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
let geek1 = tf.tensor1d([3, 4]);
let geek2 = tf.tensor2d([[3, 4], [5, 6]]);
  
tf.dot(geek1, geek2).print();

輸出：

Tensor
    [29, 36]

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

相關用法

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