Función Tensorflow.js tf.maxPool3d()

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.maxPool3d() se utiliza para calcular la agrupación máxima 3D.

Sintaxis:

tf.maxPool3d (x, filterSize, strides, pad, 
                   dimRoundingMode?, dataFormat?)

Parámetros: Esta función acepta un parámetro que se ilustra a continuación:

• x: Esto especifica el tensor de entrada de rango 5 o rango 4.
• filterSize: Esto especifica filterDepth, filterHeight, filterWidth. Si el filterSize especificado es un solo número, filterWidth == filterHeight == filterDepth.
• zancadas: especifica las zancadas de la agrupación: [strideDepth, strideHeight, strideWidth]. Si los pasos especificados son un solo número, entonces strideWidth == strideHeight ==strideDepth .
• pad: Esto especifica el tipo de algoritmo de relleno.
• dimRoundingMode: esto es opcional. Esto especifica una string de: ‘ceil’, ‘round’, ‘floor’. Si no se proporciona nada, su valor predeterminado será truncado.
• formato de datos: Esto es opcional. Esto especifica el formato de datos de los datos de entrada y salida.

Valor devuelto: Devuelve la agrupación 3D max de los elementos del tensor.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor of some elements
let pool = tf.tensor5d([11, 12, 13, 14], [2, 1, 1, 1, 2]);
  
// Calling the .maxPool3d()
let output = tf.maxPool3d(pool, 1, 2, 'valid');
  
// Printing the output.
output.print();

Producción:

Tensor
    [ [ [ [[11, 12],]]],

      [ [ [[13, 14],]]]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Initializing a tensor of some elements
let maxPool = tf.tensor5d([51, 52, 53, 54, 55, 56, 57, 58], [1, 2, 2, 2, 1]);
  
// Calling the .avgPool3d() function and
 // printing the result.
tf.avgPool3d(maxPool, 2, 1, 'valid').print();

Producción:

Tensor
     [ [ [ [[54.5],]]]]

Referencia: https://js.tensorflow.org/api/3.6.0/#maxPool3d