Tensorflow.js tf. Clase variable .assign() método

TensorFlow.js es una biblioteca JavaScript de código abierto diseñada por Google para desarrollar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo. La clase tf.Variable se usa para crear, rastrear y administrar variables en Tensorflow. Las variables de Tensorflow representan los tensores cuyos valores se pueden cambiar ejecutando operaciones en ellos.

La asignación() es el método disponible en la clase Variable que se utiliza para asignar el nuevo tf.Tensor a la variable. El nuevo valor debe tener la misma forma y tipo que el antiguo valor de la variable.

Sintaxis:

assign(newValue)

Parámetros:

• newValue: Es un objeto de tipo tf.Tensor. Este nuevo valor se asigna a la variable.

Valor devuelto: Devuelve nulo.

Ejemplo 1: este ejemplo primero crea un nuevo objeto tf.Variable con un valor inicial y luego asigna a esta variable un nuevo valor con la misma forma y tipo de valor inicial.

// 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()

Producción:

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

Ejemplo 2: este ejemplo primero crea un nuevo objeto tf.Variable con un valor inicial y luego intenta asignar a esta variable un nuevo valor con una forma diferente como valor inicial. Esto dará un error.

// 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()

Producción:

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

Referencia: https://js.tensorflow.org/api/latest/#tf.Variable.assign