O tf.Função de pilha no tensorflow.JS

No tensorflow.Biblioteca JS, o TF.A função Stack () é usada para se juntar a dois ou mais tensores.

Sintaxe:

Parâmetros:

1. São necessários dois ou mais tensores como um parâmetro que pode ser unidimensional como o primeiro parâmetro.
2. O eixo leva dois valores possíveis (0 e 1).

Se especificado 0, tf.Stack () se junta aos tensores um após o outro retornando um novo tensor.

Se especificado 1, tf.Stack () se une.

Exemplo 1

Crie dois tensores 1D com números inteiros e empilhe dois tensores configurando o eixo-0.

Saída:

Tensor1 e tensor2 estão empilhados um após o outro.

Exemplo 2

Crie dois tensores 1D com números inteiros e empilhe dois tensores definindo o eixo-1.

Saída:

Tensor1 e Tensor2 são o elemento empilhado por elemento em ambos os tensores, e o tensor retornado tem elementos empilhados colocados em linhas separadas.

Exemplo 3

Crie quatro tensores 1D com números inteiros e empilhá-los definindo o eixo-1 e o eixo-0 separadamente.

Saída:

Na primeira saída, quatro tensores são empilhados por elemento e na segunda saída, os tensores são empilhados por vez.

Conclusão

Neste tensorflow.JS Tutorial, vimos como empilhar dois ou mais tensores usando o TF.função pilha (). Se o eixo for especificado como 0, tf.Stack () se junta aos tensores um após o outro retornando um novo tensor. Se especificado como 1, tf.Stack () se une. Certifique-se de entender todos os exemplos discutidos e obtenha a diferença entre os parâmetros Axis-0/1.