Tensorflow.js tf.variable()函数

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它帮助开发人员使用 JavaScript 开发 ML 模型，并直接在浏览器或 Node.js 中使用 ML。

tf.variable()函数用于使用提供的初始值创建一个新变量。

句法：

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

参数：

• initialValue：将初始化新变量的张量的初始值。
• 可训练：它是一个可选参数。如果允许真正的优化器更新变量并且不允许错误的优化器更新它，则它是布尔类型。
• name：也是一个可选参数。它是字符串类型的。它用于用作唯一 ID 的变量的名称。
• dtype：它也是一个可选参数。如果它作为参数传递，则 intialValue 将更改为指定的 dtype。

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

示例 1：

Javascript

// Creating and initializing 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：

Javascript

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

输出：

Tensor
   [1, 2, 5, 6]

示例 3：

Javascript

// Creating and initializing 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