Función Tensorflow tf.mul()

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

Sintaxis:

tf.mul( a, b )

Parámetros:

• a: Contiene el primer objeto tf.Tensor que se multiplicó por el segundo objeto tf.Tensor por elementos. El valor de este parámetro puede ser tf.TensorTypedArray|Array.
• b: Contiene el segundo objeto tf.Tensor que se multiplicó por el 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 mul() function to multiply
// two Tensor objects
arr1.mul(arr2).print();

Producción:

Tensor
    [50, 200, 450, 800, 1250]

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 mul() function to multiply
// two Tensor objects
arr1.mul(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([5, 10, 15, 20, 25]);
 
// Declare a number
const num = tf.scalar(30);
 
// Use mul() function to multiply
// Tensor object and a number
arr.mul(num).print();

Producción:

Tensor
    [150, 300, 450, 600, 750]

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