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

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.concatenate() se usa para concatenar una array de entradas.

Sintaxis:

tf.layers.concatenate()

Parámetros:

  • args(Objeto): especifica el objeto dado. Es un parámetro opcional.
  • eje (número): especifica el eje a lo largo del cual se concatenarán las entradas. Sigue una indexación basada en 0 y su valor debe estar en el rango [-rango, rango). Puede ser tanto positivo como negativo. Aquí, el eje positivo varía de 0 a rango (valores) y especifica la dimensión del eje. El valor negativo especifica eje + rango (valores)-ésima dimensión. Puede ser tanto positivo como negativo. El valor predeterminado es 0.
  • 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 no se ha especificado aún.
  • 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.
  • entrenable: especifica si los pesos de esta capa son actualizables por ajuste.
  • pesos: especifica los valores de peso iniciales de la capa.
  • inputDType: ‘float32’ o ‘int32’ o ‘bool’ o ‘complex64’ o ‘string’.

Valor de retorno: un solo tensor, que es la concatenación de todas las entradas.

Ejemplo 1:

Javascript

// Import the library
import * as tf from "@tensorflow/tfjs"
 
// Inputs
const input1 = tf.input({shape: [3, 2]})
const input2 = tf.input({shape: [3, 2]})
const input3 = tf.input({shape: [3, 2]})
 
// Create new concatenate layer
const concatenateLayer = tf.layers.concatenate();
const output = concatenateLayer.apply([input1, input2, input3]);
 
// Print shape of resulting tensor
console.log(JSON.stringify(output.shape));

 Producción:

[pre][null, 3, 6] 

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 new concatenate layer
const concatLayer = tf.layers.concatenate();
const output = concatLayer.apply([input1, input2, input3]);
 
// Print resulting tensor
console.log(output);

 
 Producción:

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

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

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 *