Función Tensorflow.js tf.linalg.bandPart()

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. También ayuda a los desarrolladores a desarrollar modelos ML en lenguaje JavaScript y puede usar ML directamente en el navegador o en Node.js.

La función tf.linalg.bandPart() se usa para calcular la parte de la banda del tensor dado.

Sintaxis :

tf.linalg.bandPart (a, numLower, numUpper)

Parámetros:

• a: es tf.tensor a pasar.
• numLower: Número de subdiagonales a conservar. Si es negativo, mantenga todo el triángulo inferior
• numUpper: Número de subdiagonales a conservar. Si es negativo, mantenga todo el triángulo superior

Valor devuelto: Devuelve tf.Tensor1D.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor2d([[ 0,  1,  2, 3],
                     [ 4,  0,  1, 2],
                     [ 3,  1,  0, 1],
                     [ 3,  1,  1, 0]]);
  
let y = tf.linalg.bandPart(x, 1, 3);
y.print();

Producción:

Tensor
   [[0, 1, 2, 3],
    [4, 0, 1, 2],
    [0, 1, 0, 1],
    [0, 0, 1, 0]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
const x = tf.tensor2d([[ 0,  1,  2, 3],
                     [ 4,  0,  1, 2],
                     [ 3,  1,  0, 1],
                     [ 3,  1,  1, 0]]);
  
let k = tf.linalg.bandPart(x, -1, -3);
k.print();

Producción:

Tensor
   [[0, 1, 2, 3],
    [4, 0, 1, 2],
    [3, 1, 0, 1],
    [3, 1, 1, 0]]

Referencia: https://js.tensorflow.org/api/latest/#linalg.bandPart