Función Tensorflow tf.sub()

La función tf.sub() devuelve la resta de dos elementos de objetos tf.Tensor. El objeto tf.Tensor representa la array multidimensional de números.

Sintaxis:

tf.sub( a, b )

Parámetros:

• a: Contiene el primer objeto tf.Tensor. El valor de este parámetro puede ser tf.TensorTypedArray|Array.
• b: Contiene el segundo objeto tf.Tensor que resta del primer objeto tf.Tensor. El valor de este parámetro puede ser (tf.Tensor|TypedArray|Array). El tipo de este parámetro es el mismo que el tipo de a.

Valor devuelto: esta función devuelve el objeto tf.Tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Declare the Tensor array
const arr1 = tf.tensor1d([10, 20, 30, 40, 50]);
const arr2 = tf.tensor1d([5, 10, 15, 20, 25]);
 
// Use sub() function to subtract
// two Tensor objects
arr1.sub(arr2).print();

Producción:

Tensor
    [5, 10, 15, 20, 25]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Declare the Tensor array
const arr1 = tf.tensor1d([30, 40, 50]);
const arr2 = tf.tensor1d([5, 10, 15, 20, 25]);
 
// Use sub() function to subtract
// two Tensor objects
arr1.sub(arr2).print();

Producción:

An error occurred on line: 7
Operands could not be broadcast together with shapes 3 and 5.

Ejemplo 3:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Declare the Tensor array
const arr = tf.tensor1d([15, 10, 25, 20, 35]);
 
// Declare a number
const num = tf.scalar(30);
 
// Use sub() function to subtract number
// from Tensor object
arr.sub(num).print();

Producción: 

Tensor
    [-15, -20, -5, -10, 5]

Referencia: https://js.tensorflow.org/api/latest/#sub