Tensorflow.js tf. Clase TensorBuffer

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

Este objeto mutable de la clase TensorBuffer es similar a tf.Tensor , que permite a los usuarios establecer valores en las ubicaciones especificadas antes de convertirlo en un tf.Tensor inmutable. Para crear un búfer de tensor, se usa tf.buffer(). 

Esta clase TensorBuffer tiene tres funciones incorporadas que se ilustran a continuación: 

• tf.TensorBuffer clase .set() función
• tf.TensorBuffer clase .get() función
• tf.TensorBuffer clase .toTensor() función

La función .set() de la clase tf.TensorBuffer se usa para establecer un valor dado en el búfer en una ubicación específica.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 2*2 dimensions
const buffer = tf.buffer([2, 2]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 1, 0); 
  
// Converting the above buffer
// back to a tensor value to print
buffer.toTensor().print();

Producción:

Tensor
  [[5 , 0],
   [10, 0]]

La función tf.TensorBuffer class .get() se usa para devolver el valor en el búfer especificado para la ubicación dada.

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 2*2 dimensions
const buffer = tf.buffer([2, 2]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 1, 0); 
  
// Getting the values for the
// specified indices with the
// help of .get() function
console.log(buffer.get(0, 0));
console.log(buffer.get(1, 0));

Salida: La función tf.TensorBuffer class .toTensor() se usa para crear un objeto Tensor inmutable a partir del búfer especificado y sus valores.

5
10 

Ejemplo 3:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating a buffer of 2*2 dimensions
const buffer = tf.buffer([2, 2]); 
  
// Setting values at particular indices. 
buffer.set(5, 0, 0); 
buffer.set(10, 1, 0); 
  
// Converting the above buffer
// back to a tensor value to print
// with the help of .toTensor() function
buffer.toTensor().print();

Producción:

Tensor
 [[5 , 0],

Referencia: https://js.tensorflow.org/api/latest/#class:TensorBuffer