Tensorflow.js tf.avgPool3d() Función

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.avgPool3d() se usa para calcular la agrupación promedio 3D.

Sintaxis:

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

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

• x: El tensor de entrada especificado 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 de Retorno: Devuelve el Pooling promedio 3D de los elementos del tensor.

A continuación se muestran los ejemplos que ilustran el uso de la función avgPool3d().

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Initializing a tensor of some elements
let geek = tf.tensor5d([10, 11, 12, 13, 14, 15, 16, 17],
                       [1, 2, 2, 2, 1]);
 
// Calling the .avgPool3d() function over
// the above tensor as its parameter
let outcome = tf.avgPool3d(geek, 2, 1, 'valid');
 
//Printing the result.
outcome.print();

Producción:

Tensor
     [ [ [ [[13.5],]]]]

Ejemplo 2:

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

Producción:

Tensor
  [ [ [ [[51, 52],]]],

    [ [ [[53, 54],]]]]

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