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