Il tf.Funzione stack in Tensorflow.Js

Nel flusso di tensor.Biblioteca JS, TF.La funzione stack () viene utilizzata per unirsi a due o più tensori.

Sintassi:

Parametri:

1. Ci vogliono due o più tensori come parametro che può essere monodimensionale come primo parametro.
2. L'asse prende due possibili valori (0 e 1).

Se specificato 0, TF.Stack () si unisce ai tensori uno dopo l'altro restituendo un nuovo tensore.

Se specificato 1, TF.stack () si unisce ad elemento per elemento di fila restituendo un nuovo tensore.

Esempio 1

Crea due tensori 1D con numeri interi e impila due tensori impostando Axis-0.

Produzione:

Tensor1 e Tensor2 sono impilati uno dopo l'altro.

Esempio 2

Crea due tensori 1D con numeri interi e impila due tensori impostando Axis-1.

Produzione:

Tensor1 e tensore2 sono elementi impilati per elemento in entrambi i tensori e il tensore restituito ha elementi impilati posizionati in file separate.

Esempio 3

Crea quattro tensori 1D con numeri interi e impilarli impostando separatamente Axis-1 e Axis-0.

Produzione:

Nel primo output, quattro tensori sono impilati per elemento e nella seconda uscita, i tensori sono impilati alla volta.

Conclusione

In questo tensorflow.JS Tutorial, abbiamo visto come impilare due o più tensori usando il TF.funzione stack (). Se l'asse è specificato come 0, TF.Stack () si unisce ai tensori uno dopo l'altro restituendo un nuovo tensore. Se specificato come 1, TF.stack () si unisce ad elemento per elemento di fila restituendo un nuovo tensore. Assicurati di comprendere tutti gli esempi discussi e ottenere la differenza tra i parametri Axis-0/1.