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

Tensorflow.js是Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。它可以幫助開發人員使用JavaScript開發ML模型，並直接在瀏覽器或Node.js中使用ML。

tf.variable()函數用於創建具有提供的初始值的新變量。

用法:

tf.variable(initialValue, trainable, name, dtype)

參數：

• initialValue:從中初始化新變量的張量的初始值。
• trainable:它是一個可選參數。如果允許真正的優化器更新變量，而不允許偽造的優化器更新變量，則它是布爾類型。
• name:它也是一個可選參數。它是字符串類型。它用作用作唯一ID的變量的名稱。
• dtype:它也是一個可選參數。如果將其作為參數傳遞，則intialValue將更改為指定的dtype。

返回值：此函數返回tf.variable。

範例1：

// Creating and intializing a new variable 
var val = tf.variable(tf.tensor2d(
    [8, 2, 5, 6], 
    [2, 2]
));
  
// Printing the tensor
val.print()

輸出：

Tensor
   [[8, 2],
    [5, 6]]

範例2：

// Creating and intializing a new variable 
var val = tf.variable(tf.tensor([1, 2, 5, 6]));
  
// Printing the tensor
val.print()

輸出：

Tensor
   [1, 2, 5, 6]

範例3：

// Creating and intializing a new variable 
const x = tf.variable(tf.tensor([1, 2, 3]),
    true,"gfg",'complex64');
  
// Printing the tensor
x.print();

輸出：

Tensor
   [1 + 0j, 2 + 0j, 3 + 0j]

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