El TF.Función de pila en TensorFlow.Js

En el flujo de tensor.Biblioteca JS, la TF.La función stack () se usa para unir dos o más tensores.

Sintaxis:

Parámetros:

1. Se necesitan dos o más tensores como parámetro que puede ser unidimensional como el primer parámetro.
2. El eje toma dos valores posibles (0 y 1).

Si se especifica 0, TF.stack () se une a los tensores uno tras otro devolviendo un nuevo tensor.

Si se especifica 1, TF.stack () se une al elemento por elemento consecutivo devolviendo un nuevo tensor.

Ejemplo 1

Cree dos tensores 1D con enteros y apile dos tensores configurando el eje-0.

Producción:

Tensor1 y tensor2 se apilan uno tras otro.

Ejemplo 2

Cree dos tensores 1D con enteros y apile dos tensores estableciendo el eje-1.

Producción:

Tensor1 y tensor2 están apilados elementos por elemento en ambos tensores, y el tensor devuelto ha apilado elementos colocados en filas separadas.

Ejemplo 3

Cree cuatro tensores 1D con enteros y apílalos estableciendo el eje-1 y el eje-0 por separado.

Producción:

En la primera salida, cuatro tensores están apilados elementos por elemento, y en la segunda salida, los tensores se apilan a la vez.

Conclusión

En este flujo tensor.JS Tutorial, vimos cómo apilar dos o más tensores usando el TF.función stack (). Si el eje se especifica como 0, TF.stack () se une a los tensores uno tras otro devolviendo un nuevo tensor. Si se especifica como 1, TF.stack () se une al elemento por elemento consecutivo devolviendo un nuevo tensor. Asegúrese de comprender todos los ejemplos discutidos y obtener la diferencia entre los parámetros del eje-0/1.