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