Función Tensorflow.js tf.layers.multiply()

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.

La función tf.layers.multiply() se utiliza para realizar la multiplicación por elementos de una array de entradas.

Sintaxis:

tf.layers.multiply()

Parámetros:

  • inputShape: si se define este parámetro, creará otra capa de entrada para insertar antes de esta capa.
  • batchInputShape: si se define este parámetro, creará otra capa de entrada para insertar antes de esta capa.
  • batchSize: se usa para construir batchInputShape, si aún no se ha especificado.
  • dtype: especifica el tipo de datos para esta capa. El valor predeterminado de este parámetro es ‘float32’.
  • nombre: especifica el nombre de esta capa.
  • actualizable: especifica si los pesos de esta capa se pueden actualizar por ajuste o no.
  • entrenable: especifica si los pesos de esta capa son actualizables por ajuste.
  • pesos: especifica los valores de peso iniciales de la capa.
  • i nputDType: ‘float32’ o ‘int32’ o ‘bool’ o ‘complex64’ o ‘string’.

Valor devuelto: tensor único del mismo tipo que los tensores de entrada.

Ejemplo 1:

Javascript

// Import the library
import * as tf from "@tensorflow/tfjs"
 
const input1 = tf.input({shape: [3, 2]})
const input2 = tf.input({shape: [3, 2]})
const input3 = tf.input({shape: [3, 2]})
 
// Create a multiply layer
const multiplyLayer = tf.layers.multiply()
 
// Multiple array of inputs by applying multiplyLayer
const product = multiplyLayer.apply([input1, input2, input3])
 
// Print the shape of output tensor
console.log(JSON.stringify(product.shape))

Producción:

[null,3,2]

Nota: Aquí nulo denota el tamaño de lote indeterminado.

Ejemplo 2:

Javascript

// Import the library
import * as tf from "@tensorflow/tfjs"
 
// Inputs
const input1 = tf.tensor([-2, 1, 0, 5]);
const input2 = tf.tensor([3, 2, 3, 2]);
const input3 = tf.tensor([4, 3, 1, 2]);
 
// Create multiply layer
const multiplyLayer = tf.layers.multiply();
 
// Multiply inputs
const product = multiplyLayer.apply(
    [input1, input2, input3]);
 
// Print product
console.log(product);

Producción:

Tensor
    [-24, 6, 0, 20]

Referencia: https://js.tensorflow.org/api/latest/#layers.multiply

Publicación traducida automáticamente

Artículo escrito por abhinavjain194 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *