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

Tensorflow.js es un conjunto de herramientas de código abierto desarrollado por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el navegador o en la plataforma del Node. También permite a los desarrolladores crear modelos de aprendizaje automático en JavaScript y utilizarlos directamente en el navegador o con Node.js.

La función tf.layers.softmax() se usa para aplicar la función de activación softmax en los datos.

Sintaxis:

 tf.layers.softmax(args?)

Forma de entrada: arbitraria. Cuando utilice esta capa como capa inicial en un modelo, use la configuración inputShape.

Forma de salida: la salida tiene la misma forma que la entrada.

Parámetros: Acepta el objeto args el cual puede tener las siguientes propiedades:

• eje: es un número entero, la normalización softmax se realiza a lo largo de este eje. Su valor predeterminado es -1.
• inputShape: si se establece esta propiedad, se utilizará para construir una capa de entrada que se insertará antes de esta capa.
• batchInputShape: si se establece esta propiedad, se creará e insertará una capa de entrada antes de esta capa.
• batchSize: si no se proporciona batchInputShape y sí inputShape, se utiliza batchSize para crear el batchInputShape.
• dtype: Es el tipo de tipo de datos para esta capa. float32 es el valor predeterminado. Este parámetro se aplica exclusivamente a las capas de entrada.
• nombre: Este es el nombre de la capa y es de tipo string.
• entrenable: si los pesos de esta capa se pueden cambiar por ajuste. Verdadero es el valor predeterminado.
• pesos: Los valores de peso iniciales de la capa.

Devoluciones: Devuelve un objeto (Softmax).

Ejemplo 1:

import * as tf from "@tensorflow/tfjs";
  
const softmaxLayer = tf.layers.softmax({axis: 0});
const x = tf.tensor([1, 2, 1]);
  
softmaxLayer.apply(x).print();

Producción:

Tensor
   [0.2119416, 0.5761169, 0.2119416]

Ejemplo 2:

import * as tf from "@tensorflow/tfjs";
  
const softmaxLayer = tf.layers.softmax({axis: 0});
const x = tf.tensor([11, 15, 10, 12]);
  
softmaxLayer.apply(x).print();

Producción:

Tensor
   [0.0170403, 0.9303705, 0.0062688, 0.0463204]

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