Tensorflow.js tf.Variable.assign()用法及代码示例

TensorFlow.js 是一个由 Google 设计的开源 JavaScript 库，用于开发机器学习模型和深度学习神经网络。 tf.Variable 类用于在 Tensorflow 中创建、跟踪和管理变量。 Tensorflow 变量表示可以通过对其运行操作来更改其值的张量。

assign() 是 Variable 类中可用的方法，用于将新的 tf.Tensor 分配给变量。新值必须与旧变量值具有相同的形状和数据类型。

用法:

assign(newValue)

参数：

newValue:它是一个 tf.Tensor 类型的对象。这个新值被分配给变量。

返回值：它返回void。

范例1：此示例首先创建一个具有初始值的新 tf.Variable 对象，然后为该变量分配一个与初始值具有相同形状和数据类型的新值。

Javascript


// Defining tf.Tensor object for initial value
initialValue = tf.tensor([[1, 2, 3]])
  
// Defining tf.Variable object 
let x = new tf.Variable(initialValue);
  
// Printing variables dtype
console.log("dtype:",x.dtype)
  
// Printing variable shape
console.log("shape:",x.shape)
  
// Printing the tf.Variable object
x.print()
  
// Defining new tf.Tensor object of same 
// shape and dtype as initial tf.Tensor
newValue = tf.tensor([[5, 8, 10]])
  
// Assigning new value to the variable
x.assign(newValue)
  
// Printing the tf.Variable object
x.print()

输出：

dtype:float32
shape:1,3
Tensor
     [[1, 2, 3],]
Tensor
     [[5, 8, 10],]

范例2：此示例首先创建一个具有初始值的新 tf.Variable 对象，然后尝试为该变量分配一个具有不同形状的新值作为初始值。这将给出一个错误。

Javascript


// Defining tf.Tensor object for initial value
initialValue = tf.tensor([[1, 2],[3, 4]])
  
// Defining tf.Variable object 
let x = new tf.Variable(initialValue);
  
// Printing variables dtype
console.log("dtype:",x.dtype)
  
// Printing variable shape
console.log("shape:",x.shape)
  
// Printing the tf.Variable object
x.print()
  
// Defining new tf.Tensor object of same 
// shape as initial tf.Tensor
newValue = tf.tensor([[5, 6],[10, 11]])
  
// Assigning new value to the variable
x.assign(newValue)
  
// Printing the tf.Variable object
x.print()

输出：

dtype:float32
shape:2,2
Tensor
    [[1, 2],
     [3, 4]]
Tensor
    [[5 , 6 ],
     [10, 11]]

参考：https://js.tensorflow.org/api/latest/#tf.Variable.assign

相关用法

注：本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Tensorflow.js tf.Variable class .assign() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。