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

Tensorflow.js 是 Google 開發的一個開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。 Tensorflow.js tf.layers.dot() 函數用於在提供的兩個張量之間應用點積。

用法：

tf.layers.dot(args);

參數：該函數接受以下參數：

args:它是具有以下字段的對象：
- axes:它定義了點積的坐標軸。
- normalize:它是用於以餘弦值形式生成點積答案的布爾數據。
- inputShape:它定義了模型輸入層的形狀。它用於創建輸入層。
- dtype:它是圖層的數據類型。它用於模型的第一層。
- name:字符串是輸入層的名稱。
- weight:它是張量，其值為層的初始值。
- trainable:它定義了層是否可訓練。它是一種布爾數據類型。
- batchInputShape:它為輸入層中的樣本定義了批次的形狀。它用於製作輸入層。
- batchSize:在輸入層的構建中作為batchInputShape的補充。它用於製作輸入層。
- inputDType:它是層中輸入數據的數據類型。

返回值：它返回兩個張量的乘積張量。

以下是此函數的一些示例。

範例1：在這個例子中，我們將簡單地製作帶有 -2 軸旋轉值的張量的點積。

Javascript


// import * as tf from "@tensorflow/tfjs"
  
// Generating tensor of [3,3] shape and size.
const geek_tens1 = tf.randomUniform([3 ,3]);
const geek_tens2 = tf.randomUniform([3 ,3]);
  
// Making tensor campatiable to apply to dot product.
const geek_Input = [geek_tens1 ,geek_tens2];
  
// Calling dot product
const geek_config = {axes:-2};
const geek_DotPro = tf.layers.dot(geek_config);
const geek_DotProRes = geek_DotPro.apply(geek_Input);
  
// Printing our result
console.log(geek_DotProRes);
geek_DotProRes.print();

輸出：

Tensor
    [[0.7034817],
     [0.2338114],
     [1.1493738]]
Tensor
    [[0.7034817],
     [0.2338114],
     [1.1493738]]

範例2：在這個例子中，我們將製作兩個張量的點積，並以餘弦形式查看結果。

Javascript


import * as tf from "@tensorflow/tfjs"
  
// Collect both outputs and print separately.
// Generating tensor of [3,3] shape and size.
const geek_tens1 = tf.randomUniform([2 ,4]);
const geek_tens2 = tf.randomUniform([2 ,4]);
  
// Making tensor campatiable to apply to dot product.
const geek_Input = [geek_tens1 ,geek_tens2];
  
// Calling dot product
const geek_config = {axis:1, normalize:true,
        dtype:'int32', name:'DotProduct' };
const geek_DotPro = tf.layers.dot(geek_config);
const geek_DotProRes = geek_DotPro.apply(geek_Input);
  
// Printing our result
console.log(geek_DotProRes);

輸出：

Tensor
    0.6817111968994141

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

相關用法

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